Function cast

test/mockgpu/amd/pcode.py:133–133

(v: UOp)

def _minmax_reduce(is_max: bool, dt, *args: UOp) -> UOp:
  def cast(v: UOp) -> UOp: return v.bitcast(dt) if dt == dtypes.float32 and v.dtype == dtypes.uint32 else v.cast(dt)
  def minmax(a: UOp, b: UOp) -> UOp:
    if dt in (dtypes.uint8, dtypes.uint16, dtypes.uint32, dtypes.uint64): return (a > b).where(a, b) if is_max else (a < b).where(a, b)
    return a.maximum(b) if is_max else a.minimum(b)