MCPcopy
hub / github.com/tinygrad/tinygrad / end

Method end

tinygrad/uop/ops.py:509–509  ·  view source on GitHub ↗
(self, *src:UOp)

Source from the content-addressed store, hash-verified

507 def wait(self, src:UOp|ConstType, **kwargs):
508 return UOp(Ops.WAIT, dtypes.void, (self, self.const_like(src) if not isinstance(src, UOp) else src), **kwargs)
509 def end(self, *src:UOp): return UOp(Ops.END, src=(self,)+src) if len(src) else self
510 def after(self, *src:UOp, **kwargs): return UOp(Ops.AFTER, self.dtype, (self,)+src, **kwargs) if len(src) else self
511 def barrier(self, *src:UOp): return UOp(Ops.BARRIER, src=(self,)+src)
512 def ins(self, arg, **kwargs): return UOp(Ops.INS, kwargs.pop("dtype", self.dtype), kwargs.pop("src", self.src), arg, kwargs.pop("tag", self.tag))

Callers 15

_embedding_bwd_kernelFunction · 0.95
dcache_flushFunction · 0.95
encodeMethod · 0.45
_zero_kernelFunction · 0.45
setMethod · 0.45
expander.pyFile · 0.45
do_split_endsFunction · 0.45
reduce_to_accFunction · 0.45
merge_reduce_endsFunction · 0.45
add_ranges_to_storeFunction · 0.45
rangeify.pyFile · 0.45
bufferize_to_storeFunction · 0.45

Calls 1

UOpClass · 0.85

Tested by 15

test_global_prod_maxMethod · 0.36
test_after_endMethod · 0.36
fxnMethod · 0.36
test_fail_1Method · 0.36
custom_arange_kernelFunction · 0.36
custom_eye_kernelFunction · 0.36
custom_add_one_kernelFunction · 0.36