Create a lane range UOp with unique axis ID.
(self, n: int | None = None)
| 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).""" |
no outgoing calls