MCPcopy Index your code
hub / github.com/tinygrad/tinygrad / mstack

Method mstack

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

Source from the content-addressed store, hash-verified

650 return UOp(Ops.COPY, self.dtype, (inp, UOp(Ops.DEVICE, arg=device) if not isinstance(device, UOp) else device))
651 def mselect(self, arg:int) -> UOp: return UOp(Ops.MSELECT, self.dtype, (self,), arg)
652 def mstack(self, *srcs: UOp) -> UOp: return UOp(Ops.MSTACK, self.dtype, (self,)+srcs)
653 @property
654 def metadata(self) -> tuple[Metadata, ...]|None: return all_metadata.get(self, None)
655

Callers 2

_multi_likeMethod · 0.80
handle_allreduceFunction · 0.80

Calls 1

UOpClass · 0.85

Tested by

no test coverage detected