MCPcopy
hub / github.com/tinygrad/tinygrad / range

Method range

test/mockgpu/amd/emu.py:445–449  ·  view source on GitHub ↗

Create a lane range UOp with unique axis ID.

(self, n: int | None = None)

Source from the content-addressed store, hash-verified

443 self.accvgpr = self.vgpr
444
445 def range(self, n: int | None = None) -> UOp:
446 """Create a lane range UOp with unique axis ID."""
447 if n is None: n = self.wave_size
448 self._axis_id += 1
449 return UOp.range(n, self._axis_id, AxisType.LOOP, dtype=dtypes.int)
450
451 def unroll_lanes(self, get_lane_bit, exec_mask: UOp, apply_exec: bool = True) -> UOp:
452 """Combine lane bits into a mask using RANGE+REDUCE (32-bit for RDNA, 64-bit for CDNA)."""

Calls

no outgoing calls