MCPcopy
hub / github.com/tinygrad/tinygrad / schedule_linear

Method schedule_linear

tinygrad/tensor.py:234–238  ·  view source on GitHub ↗

Creates the schedule needed to realize these Tensor(s).

(self, *lst:Tensor)

Source from the content-addressed store, hash-verified

232 return create_linear_with_vars(big_sink)
233
234 def schedule_linear(self, *lst:Tensor) -> UOp:
235 """Creates the schedule needed to realize these Tensor(s)."""
236 linear, var_vals = self.linear_with_vars(*lst)
237 assert len(var_vals) == 0
238 return linear
239
240 @disable_gc()
241 def realize(self, *lst:Tensor, do_update_stats=True) -> Tensor:

Callers 15

test_recursive_addMethod · 0.95
test_recursive_padMethod · 0.95
test_matmulFunction · 0.80
test_matmulFunction · 0.80
tvm_gemm.pyFile · 0.80
amd_matmul.pyFile · 0.80
simple_matmul.pyFile · 0.80
test_tc_upMethod · 0.80
test_max_upMethod · 0.80

Calls 1

linear_with_varsMethod · 0.95

Tested by 15

test_recursive_addMethod · 0.76
test_recursive_padMethod · 0.76
test_matmulFunction · 0.64
test_matmulFunction · 0.64
test_tc_upMethod · 0.64
test_max_upMethod · 0.64
test_mutate_addMethod · 0.64
test_schedule_gcMethod · 0.64
test_codegen_tracingMethod · 0.64