算子列表#
- abs
- abs_
- add
- add_
- addcdiv
- addcmul
- addmm
- addmv
- addr
- all
- allclose
- amax
- angle
- any
- apply_rotary_pos_emb
- arange
- argmax
- argmin
- atan
- atan_
- avg_pool2d
- batch_norm
- bitwise_and
- bitwise_and_
- bitwise_left_shift
- bitwise_not
- bitwise_not_
- bitwise_or
- bitwise_or_
- bitwise_right_shift
- bmm
- cat
- celu
- celu_
- clamp
- clamp_
- clamp_min
- clamp_min_
- concat_and_cache_mla
- contiguous
- conv1d
- conv2d
- conv3d
- cos
- cos_
- count_nonzero
- cross_entropy_loss
- cummax
- cummin
- cumsum
- diag
- diag_embed
- diagonal
- div
- div_
- dot
- dropout
- elu
- elu_
- embedding
- eq
- erf
- erf_
- exp
- exp2
- exp2_
- exp_
- exponential_
- eye
- fill
- fill_
- flash_attention_forward
- flash_attn_varlen_func
- flash_mla
- flip
- floor_divide
- floor_divide_
- full
- full_like
- fused_add_rms_norm
- gather
- ge
- gelu
- gelu_
- gelu_and_mul
- get_scheduler_metadata
- glu
- group_norm
- gt
- hstack
- index_add
- index_add_
- index_put
- index_put_
- index_select
- instance_norm
- isclose
- isfinite
- isin
- isinf
- isnan
- kron
- layer_norm
- le
- lerp
- lerp_
- linspace
- log
- log_sigmoid
- log_softmax
- logical_and
- logical_not
- logical_or
- logical_xor
- logspace
- lt
- masked_fill
- masked_fill_
- masked_select
- max
- max_pool2d
- maximum
- mean
- min
- minimum
- mm
- moe_align_block_size
- mse_loss
- mul
- mul_
- multinomial
- mv
- nan_to_num
- ne
- neg
- neg_
- nll_loss
- nonzero
- normal
- ones
- ones_like
- outer
- pad
- polar
- pow
- pow_
- prod
- quantile
- rand
- rand_like
- randn
- randn_like
- randperm
- reciprocal
- reciprocal_
- relu
- relu_
- remainder
- remainder_
- repeat
- repeat_interleave
- reshape_and_cache
- reshape_and_cache_flash
- resolve_conj
- resolve_neg
- rms_norm
- rsqrt
- rsqrt_
- rwkv_mm_sparsity
- rwkv_ka_fusion
- scaled_dot_product_attention
- scatter
- scatter_
- select_scatter
- sigmoid
- sigmoid_
- silu
- silu_
- silu_and_mul
- sin
- sin_
- skip_layer_norm
- slice_scatter
- softmax
- softplus
- sort
- sqrt
- sqrt_
- stack
- std
- sub
- sub_
- sum
- tan
- tan_
- tanh
- tanh_
- threshold
- tile
- to_copy
- topk
- trace
- topk_softmax
- triu
- uniform_
- unique
- upsample_bicubic2d_aa
- upsample_nearest2d
- var_mean
- vdot
- vector_norm
- vstack
- weight_norm
- where
- zeros
- zeros_like