跳至内容

UOp

UOp ¤

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

基类: MathTrait

操作 ¤

基类: FastEnum

属性:

名称 ¤

NAME = auto()

接收端 ¤

SINK = auto()

连续 ¤

CONTIGUOUS = auto()

连续向后 ¤

CONTIGUOUS_BACKWARD = auto()

分离 ¤

DETACH = auto()

内核 ¤

KERNEL = auto()

唯一 ¤

UNIQUE = auto()

复制 ¤

COPY = auto()

缓冲区视图 ¤

BUFFER_VIEW = auto()

¤

BLOCK = auto()

块起始 ¤

BLOCKSTART = auto()

区块分叉 ¤

BLOCKFORK = auto()

块结束 ¤

BLOCKEND = auto()

重塑 ¤

RESHAPE = auto()

置换 ¤

PERMUTE = auto()

展开 ¤

EXPAND = auto()

填充 ¤

PAD = auto()

收缩 ¤

SHRINK = auto()

FLIP ¤

FLIP = auto()

展开循环 ¤

UNROLL = auto()

合约 ¤

CONTRACT = auto()

查看 ¤

VIEW = auto()

DEFINE_GLOBAL ¤

DEFINE_GLOBAL = auto()

缓冲区 ¤

BUFFER = auto()

DEFINE_VAR ¤

DEFINE_VAR = auto()

DEFINE_LOCAL ¤

DEFINE_LOCAL = auto()

DEFINE_ACC ¤

DEFINE_ACC = auto()

有效 ¤

VALID = auto()

特别 ¤

SPECIAL = auto()

空操作 ¤

NOOP = auto()

REDUCE_AXIS ¤

REDUCE_AXIS = auto()

归约 ¤

REDUCE = auto()

GEP ¤

GEP = auto()

向量化 ¤

VECTORIZE = auto()

CAT ¤

CAT = auto()

PTRCAT ¤

PTRCAT = auto()

类型转换 ¤

CAST = auto()

BITCAST ¤

BITCAST = auto()

EXP2 ¤

EXP2 = auto()

LOG2 ¤

LOG2 = auto()

正弦函数 ¤

SIN = auto()

平方根 ¤

SQRT = auto()

倒数 ¤

RECIP = auto()

取反 ¤

NEG = auto()

加载 ¤

LOAD = auto()

存储 ¤

STORE = auto()

索引 ¤

INDEX = auto()

WMMA ¤

WMMA = auto()

乘法 ¤

MUL = auto()

SHL ¤

SHL = auto()

右移 ¤

SHR = auto()

IDIV ¤

IDIV = auto()

加法 ¤

ADD = auto()

最大值 ¤

MAX = auto()

MOD ¤

MOD = auto()

CMPLT ¤

CMPLT = auto()

不等于比较 ¤

CMPNE = auto()

异或 ¤

XOR = auto()

¤

OR = auto()

¤

AND = auto()

THREEFRY ¤

THREEFRY = auto()

减法 ¤

SUB = auto()

FDIV ¤

FDIV = auto()

幂运算 ¤

POW = auto()

WHERE条件 ¤

WHERE = auto()

乘法累加 ¤

MULACC = auto()

分配 ¤

ASSIGN = auto()

绑定 ¤

BIND = auto()

屏障 ¤

BARRIER = auto()

范围 ¤

RANGE = auto()

如果 ¤

IF = auto()

结束范围 ¤

ENDRANGE = auto()

ENDIF ¤

ENDIF = auto()

VCONST ¤

VCONST = auto()

常量 ¤

CONST = auto()

设备 ¤

DEVICE = auto()

多线程 ¤

MULTI = auto()

自定义 ¤

CUSTOM = auto()

自定义 ¤

CUSTOMI = auto()

忽略 ¤

IGNORE = auto()

FUSE ¤

FUSE = auto()