MCPcopy
hub / github.com/tinygrad/tinygrad / set

Method set

tinygrad/uop/ops.py:1003–1004  ·  view source on GitHub ↗
(self:UOp, val:UOp|ConstType, end:UOp|tuple[UOp, ...]|list[UOp]=())

Source from the content-addressed store, hash-verified

1001
1002 # set is store+end+after
1003 def set(self:UOp, val:UOp|ConstType, end:UOp|tuple[UOp, ...]|list[UOp]=()) -> UOp:
1004 return self.src[0].after(self.store(val).end(*argfix(end)))
1005
1006 # TODO: this should replace placeholder
1007 @staticmethod

Callers 15

amx.pyFile · 0.45
hand_spec_tc_coresFunction · 0.45
custom_gemmFunction · 0.45
load_to_localsFunction · 0.45
compute_on_localsFunction · 0.45
halide_gemm.pyFile · 0.45
compute_on_localsFunction · 0.45
setUpMethod · 0.45
custom_gemmFunction · 0.45
custom_sumFunction · 0.45
slice_sum_kernelFunction · 0.45
simple_qkv_kernelFunction · 0.45

Calls 4

storeMethod · 0.95
argfixFunction · 0.90
afterMethod · 0.45
endMethod · 0.45

Tested by 9

setUpMethod · 0.36
custom_gemmFunction · 0.36
custom_sumFunction · 0.36
slice_sum_kernelFunction · 0.36
simple_qkv_kernelFunction · 0.36
test_simpleMethod · 0.36
test_matmulMethod · 0.36
test_matmul_reluMethod · 0.36
custom_set0_kernelFunction · 0.36