UOp

UOp

UOp(
    op: Ops,
    dtype: DType = void,
    src: tuple[UOp, ...] = tuple(),
    arg: Any = None,
    tag: Any = None,
)

Bases: MathTrait

Ops

Bases: FastEnum

Attributes:

• NOOP –
• SINK –
• UNIQUE –
• DEVICE –
• KERNEL –
• PRECAST –
• REWRITE_ERROR –
• SENTINEL –
• AFTER –
• GROUP –
• COPY –
• BUFFER –
• BUFFER_VIEW –
• MSELECT –
• MSTACK –
• BUFFERIZE –
• CONTIGUOUS –
• CONTIGUOUS_BACKWARD –
• DETACH –
• FUSE –
• RESHAPE –
• PERMUTE –
• EXPAND –
• PAD –
• SHRINK –
• FLIP –
• MULTI –
• DEFINE_GLOBAL –
• DEFINE_LOCAL –
• DEFINE_REG –
• DEFINE_VAR –
• BIND –
• SPECIAL –
• REDUCE_AXIS –
• REDUCE –
• ALLREDUCE –
• UNROLL –
• CONTRACT –
• GEP –
• VECTORIZE –
• CAT –
• PTRCAT –
• CAST –
• BITCAST –
• EXP2 –
• LOG2 –
• SIN –
• SQRT –
• RECIP –
• NEG –
• TRUNC –
• LOAD –
• STORE –
• ASSIGN –
• WMMA –
• INDEX –
• ADD –
• MUL –
• SHL –
• SHR –
• IDIV –
• MAX –
• MOD –
• CMPLT –
• CMPNE –
• CMPEQ –
• XOR –
• OR –
• AND –
• THREEFRY –
• SUB –
• FDIV –
• POW –
• WHERE –
• MULACC –
• BARRIER –
• RANGE –
• IF –
• END –
• ENDIF –
• VCONST –
• CONST –
• CUSTOM –
• CUSTOMI –

NOOP

NOOP = auto()

SINK

SINK = auto()

UNIQUE

UNIQUE = auto()

DEVICE

DEVICE = auto()

KERNEL

KERNEL = auto()

PRECAST

PRECAST = auto()

REWRITE_ERROR

REWRITE_ERROR = auto()

SENTINEL

SENTINEL = auto()

AFTER

AFTER = auto()

GROUP

GROUP = auto()

COPY

COPY = auto()

BUFFER

BUFFER = auto()

BUFFER_VIEW

BUFFER_VIEW = auto()

MSELECT

MSELECT = auto()

MSTACK

MSTACK = auto()

BUFFERIZE

BUFFERIZE = auto()

CONTIGUOUS

CONTIGUOUS = auto()

CONTIGUOUS_BACKWARD

CONTIGUOUS_BACKWARD = auto()

DETACH

DETACH = auto()

FUSE

FUSE = auto()

RESHAPE

RESHAPE = auto()

PERMUTE

PERMUTE = auto()

EXPAND

EXPAND = auto()

PAD

PAD = auto()

SHRINK

SHRINK = auto()

FLIP

FLIP = auto()

MULTI

MULTI = auto()

DEFINE_GLOBAL

DEFINE_GLOBAL = auto()

DEFINE_LOCAL

DEFINE_LOCAL = auto()

DEFINE_REG

DEFINE_REG = auto()

DEFINE_VAR

DEFINE_VAR = auto()

BIND

BIND = auto()

SPECIAL

SPECIAL = auto()

REDUCE_AXIS

REDUCE_AXIS = auto()

REDUCE

REDUCE = auto()

ALLREDUCE

ALLREDUCE = auto()

UNROLL

UNROLL = auto()

CONTRACT

CONTRACT = auto()

GEP

GEP = auto()

VECTORIZE

VECTORIZE = auto()

CAT

CAT = auto()

PTRCAT

PTRCAT = auto()

CAST

CAST = auto()

BITCAST

BITCAST = auto()

EXP2

EXP2 = auto()

LOG2

LOG2 = auto()

SIN

SIN = auto()

SQRT

SQRT = auto()

RECIP

RECIP = auto()

NEG

NEG = auto()

TRUNC

TRUNC = auto()

LOAD

LOAD = auto()

STORE

STORE = auto()

ASSIGN

ASSIGN = auto()

WMMA

WMMA = auto()

INDEX

INDEX = auto()

ADD

ADD = auto()

MUL

MUL = auto()

SHL

SHL = auto()

SHR

SHR = auto()

IDIV

IDIV = auto()

MAX

MAX = auto()

MOD

MOD = auto()

CMPLT

CMPLT = auto()

CMPNE

CMPNE = auto()

CMPEQ

CMPEQ = auto()

XOR

XOR = auto()

OR

OR = auto()

AND

AND = auto()

THREEFRY

THREEFRY = auto()

SUB

SUB = auto()

FDIV

FDIV = auto()

POW

POW = auto()

WHERE

WHERE = auto()

MULACC

MULACC = auto()

BARRIER

BARRIER = auto()

RANGE

RANGE = auto()

IF

IF = auto()

END

END = auto()

ENDIF

ENDIF = auto()

VCONST

VCONST = auto()

CONST

CONST = auto()

CUSTOM

CUSTOM = auto()

CUSTOMI

CUSTOMI = auto()