forked from ccf-ai-infra/GPUCodeForces
583 lines
8.4 KiB
Plaintext
583 lines
8.4 KiB
Plaintext
# Operator targets tracked in README.md
|
||
# Add unimplemented operators here, one per line.
|
||
# scripts/update_operator_checklist.py checks them when matching implementations appear under S1 codes/.
|
||
|
||
aconc
|
||
ActorCriticLoss
|
||
adaptive_maxpool1d
|
||
adaptive_maxpool3d
|
||
Adaptivepiecewiselinear
|
||
AdvantageLoss
|
||
AdversarialLoss
|
||
Affine+ReLU6 融合
|
||
affine_grid3d
|
||
affineleakyreluclamp
|
||
AHAF
|
||
AHerfReLU
|
||
alpha_iou
|
||
alphablend
|
||
angle
|
||
AngularDistance
|
||
AngularLoss
|
||
AOAF
|
||
APALU
|
||
AQuLU
|
||
arcfaceloss
|
||
ArcSinh-Affine-Gate
|
||
ArcSinTanh-Affine-Gate
|
||
Arctan-Affine-Gate
|
||
ARiA2
|
||
Atan-Sigmoid-Mix-Gate
|
||
balanced_softmax_loss
|
||
ball_query
|
||
batchnorm1d
|
||
bce
|
||
bcewithlogitsloss
|
||
BehaviorCloningLoss
|
||
BellmanLoss
|
||
BetaDivergenceLoss
|
||
BhattacharyyaDistance
|
||
Bias+GELU（tanh 近似）
|
||
Bias+SiLU（Swish）融合
|
||
bilinear
|
||
BilinearGLU
|
||
Bipolar
|
||
BipolarSigmoid
|
||
black_scholes
|
||
blurpool
|
||
box_area
|
||
box_corner_to_center
|
||
box_iou
|
||
boxfilter
|
||
braycurtis
|
||
braycurtis_adaptive_triplet
|
||
bregman_divergence_gelu
|
||
BregmanDivergenceLoss
|
||
BReLU
|
||
broadcast_tensors
|
||
bucketize
|
||
BYOLLoss
|
||
canberra
|
||
canberra_focalloss
|
||
CanberraDistance
|
||
cartesian_prod
|
||
causal_mask
|
||
CELU-Affine-Gate
|
||
CenterNetLoss
|
||
Channel RMS Gate
|
||
Channel-Softmax-Affine gating
|
||
channel_permute
|
||
channelmeangate
|
||
ChannelShuffle
|
||
CharbonnierLoss
|
||
chebyshev
|
||
chebyshev_abs_square
|
||
chebyshev_hardswish
|
||
chebyshev_leakyrelu
|
||
chebyshev_sigmoid
|
||
chebyshevaffine
|
||
ChebyshevDistance
|
||
cholesky
|
||
circleloss
|
||
circularpad1d
|
||
circularpad2d
|
||
circularpad3d
|
||
clarksDistance
|
||
cmyk_to_rgb
|
||
Colu
|
||
column_stack
|
||
ComboLoss
|
||
complex_abs_angle_polar
|
||
complex_conj_mul_div
|
||
complex_exp_log_power
|
||
complex_mul
|
||
constantpad3d
|
||
contrastiveloss
|
||
conv2d
|
||
Cos-Affine-Gate
|
||
cosfaceloss
|
||
cosine
|
||
Cosine-Affine-Gate
|
||
Cosine-Square-Affine-Gate
|
||
cosine_swish_gelu
|
||
cosinedistance
|
||
cosinedistance_softmax
|
||
CosineEmbeddingLoss
|
||
cosineloss
|
||
cosinesimilarity
|
||
CoVariance
|
||
crop_resize
|
||
cross
|
||
cross_layer_norm
|
||
CrossEntropyDiceLoss
|
||
CrossEntropyLoss
|
||
CRReLU
|
||
cumsum
|
||
DecayingSineUnit
|
||
deepnorm
|
||
depthwise_conv1d
|
||
dequantize_fp4
|
||
dequantize_int8
|
||
dequantize_linear
|
||
dice
|
||
dice_bce
|
||
dice_from_2d
|
||
DiceLoss
|
||
DiceSimilarity
|
||
digitization
|
||
Dilation1d
|
||
Dilation2d
|
||
DistillationLoss
|
||
dot_mse_tanh
|
||
DoubleGLU
|
||
dropblock1d
|
||
DSiLU
|
||
DSReLU
|
||
Dual-Input Contrastive Gate
|
||
dw_transpose
|
||
dw_transpose2d
|
||
dw_transpose3d
|
||
EIS1
|
||
EIS2
|
||
EIS3
|
||
ELU-Affine-Gate
|
||
ELUGLU
|
||
embeddingbag
|
||
EMDLoss
|
||
Erf-Affine-Gate
|
||
ErfAct
|
||
ErfReLU
|
||
Erosion1d
|
||
Erosion2d
|
||
Esh
|
||
euclidean
|
||
euclidean_erfc
|
||
evonorm
|
||
Exp-Square Affine Gate
|
||
Expm1+Affine
|
||
Expm1-Sigmoid-Affine-Gate
|
||
expnormalizelog
|
||
ExponentialLinear-Affine-Gate
|
||
fake_quantize_per_channel_affine
|
||
fake_quantize_per_tensor_affine
|
||
farthest_point_sampling
|
||
FastAPLoss
|
||
FDivergenceLoss
|
||
finite_difference
|
||
fisherrao_rmsnorm
|
||
FlattenT
|
||
flip_horizontal
|
||
fma_activation
|
||
focal_eiou
|
||
FocalLoss
|
||
focalloss_fused
|
||
focalloss_labelsmoothing
|
||
focalloss_reduction
|
||
focalloss_sigmoid
|
||
focaltverskyloss
|
||
fold
|
||
fourieraffine
|
||
FPFLU
|
||
FReLU
|
||
fresnel_schlick
|
||
FTS
|
||
FunnelActivationforVisualRecognition
|
||
fused_adam_step
|
||
fused_rmsprop_step
|
||
gamma_correction
|
||
GammaDivergenceLoss
|
||
gateblendnormalize
|
||
gather_elements
|
||
Gaussian+Bias
|
||
Gaussian-Affine-Gate
|
||
gaussian_blur
|
||
gaussian_filter_2d
|
||
gaussian_pdf
|
||
GaussianNLLLoss
|
||
GDL
|
||
geglu
|
||
GELU-Affine-Gate
|
||
gelu_dropout
|
||
gempool
|
||
GeneratorLoss
|
||
giouloss
|
||
global_average_pooling
|
||
global_response_normalization
|
||
GLU（Gated Linear Unit）融合
|
||
Gompertz-Affine-Gate
|
||
gowerdistance
|
||
GradientClip
|
||
gridsample1d
|
||
Group-Softmax-Affine gating
|
||
groupnorm
|
||
GrowingCosineUnit
|
||
GTU（tanh(a) * sigmoid(b)）
|
||
GumbelCDF
|
||
GumbelPDF
|
||
hamming
|
||
hamming_gelu
|
||
hamming_relu
|
||
hamming_sigmoid
|
||
hamming_swish
|
||
hamming_xor_and
|
||
HammingDistance
|
||
hardbootstrappingloss
|
||
HardELiSH
|
||
hardmish
|
||
hardshrink
|
||
HardShrink-Affine-Gate
|
||
HardSigmoid
|
||
Hardsigmoid+Affine
|
||
HardSigmoid-Affine-Gate
|
||
HardSReLUE
|
||
HardSwish
|
||
Hardswish-Affine-Gate
|
||
hardswishgate
|
||
HardTanh
|
||
HardTanh-Affine-Gate
|
||
hardtanhgate
|
||
haversine_distance
|
||
hellinger_bhattacharyya
|
||
hellinger_gelu
|
||
HellingerDistance
|
||
hexpo
|
||
hingeembeddingloss
|
||
hingeloss
|
||
HistogramLoss
|
||
Huber-Affine-Gate
|
||
huber_loss_tukey_biweight
|
||
huberloss
|
||
HungarianLoss
|
||
image_normalize
|
||
ImitationLearningLoss
|
||
infonceloss
|
||
Instancenorm
|
||
instancenorm_dropout
|
||
instancenorm_relu
|
||
IntraClassCorrelation
|
||
inverse_lerp
|
||
InverseReinforcementLearningLoss
|
||
InverseSquare-Affine-Gate
|
||
InvMultiquadratic
|
||
iou_tanh
|
||
IOULoss
|
||
IpLU
|
||
Isigmoid
|
||
ISRLU
|
||
ISRU
|
||
ItakuraSaitoDistanceLoss
|
||
jaccard_dice_sqrt
|
||
jaccard_legendre
|
||
JaccardSimilarity
|
||
jaro_winkler_softmax
|
||
jensenshannon_groupnorm
|
||
jsdivergence
|
||
kldiv_jsdiv_swish
|
||
kldivloss
|
||
KulczynskiIndex
|
||
kullbackleibler_layernorm
|
||
l1
|
||
l1_fused
|
||
l1loss
|
||
l2_normalize
|
||
LaLU
|
||
laplacian
|
||
Laplacian-Affine-Gate
|
||
laplacian_filter
|
||
layernorm
|
||
LDAMLoss
|
||
Leaky-Swish+Bias
|
||
LeakyReGLU
|
||
LeakyReLU-Affine-Gate
|
||
LeCunTanh
|
||
legendreaffine
|
||
linear_gelu
|
||
lisht
|
||
localresponsenorm
|
||
Log1p(Square)+Bias
|
||
Log1pAbs-Affine-Gate
|
||
logbeta
|
||
LogCosh-Affine-Gate
|
||
logcoshdiceloss
|
||
LogCoshLoss
|
||
logdet
|
||
logitsigmoidshift
|
||
LogLU
|
||
LogMeanExp
|
||
LogSigmoid
|
||
LogSigmoid Affine Gate
|
||
LogSigmoid-Affine-Gate
|
||
LogSumExp
|
||
LogWeightedSumExp
|
||
lp_pool2d
|
||
lrn_simple
|
||
Lsoftmaxloss
|
||
MahalanobisDistanceLoss
|
||
manhattan
|
||
manhattan_distance_matrix
|
||
manhattan_erf
|
||
manhattan_hardswish
|
||
manhattan_leakyrelu
|
||
manhattan_mse
|
||
manhattan_relu
|
||
manhattan_sigmoid
|
||
manhattan_sqrt
|
||
manhattan_swish
|
||
manhattan_tanh
|
||
marcsinh
|
||
marginrankingloss
|
||
matmul
|
||
maxunpool1d
|
||
maxunpool2d
|
||
meanstdnormalizeclip
|
||
median_filter_3d
|
||
MElliott
|
||
MeshEdgeLoss
|
||
MetaAconC
|
||
minkowski
|
||
minkowski_contrastiveloss
|
||
minkowski_instancenorm
|
||
minkowski_relu
|
||
MinkowskiDistance
|
||
minmax_observer
|
||
minmaxscaleshift
|
||
Mish
|
||
Mish+Bias
|
||
Mish-Affine-Gate
|
||
MishB
|
||
mishglu
|
||
mixup
|
||
MMReLU
|
||
ModeSeekingLoss
|
||
ModReLU
|
||
ModSwish
|
||
mseloss
|
||
MsewithLogitLoss
|
||
mulaw_decoding
|
||
mulaw_encoding
|
||
multilabelmarginloss
|
||
multimarginloss
|
||
Multiquadratic
|
||
MutualInformation
|
||
NIPUNA
|
||
Nish
|
||
NISRLU
|
||
NLLLoss
|
||
NLReLU
|
||
optional_get_element
|
||
optional_has_element
|
||
PAA
|
||
pairwise_distance
|
||
pairwisedistance
|
||
ParametricSigmoid
|
||
PATS
|
||
PearsonCorrelation
|
||
penalizedtanh
|
||
PerceptualLoss
|
||
permute
|
||
PGELU
|
||
piecewiseaffine
|
||
PiecewiseLinearUnit
|
||
poissonnllloss
|
||
polar_to_cartesian
|
||
polaraffine
|
||
poly1crossentropy
|
||
poly1focalloss
|
||
PolyLoss
|
||
polynomial_eval
|
||
polynomialaffine
|
||
Power-Sigmoid-Affine-Gate
|
||
Power-Sigmoid-Gate
|
||
PPOLoss
|
||
PReLU
|
||
projectiveaffine
|
||
Prototype-Cosine Gate
|
||
Pserf
|
||
PseudoHuber-Affine-Gate
|
||
PSGU
|
||
PSMish
|
||
PTELU
|
||
PTLU
|
||
python
|
||
QLearningLoss
|
||
QReLU
|
||
QuantileLoss
|
||
quantilenormalizeexpand
|
||
RadialBasisFunction
|
||
Range-Gate Affine
|
||
rangescalegate
|
||
rangeshiftgate
|
||
rank_normalize_scale
|
||
ranknetloss
|
||
ransac_normalize_outlier_reject
|
||
RationalClip-Affine-Gate
|
||
RationalFunctionApproximator
|
||
real_imag_hypot
|
||
ReflectionPad3d
|
||
ReLTanh
|
||
ReLU-Affine-Gate
|
||
ReLU6-Affine-Gate
|
||
ReLU^2
|
||
RenyiDivergenceLoss
|
||
repeat_interleave
|
||
replicationpad1d
|
||
replicationpad2d
|
||
replicationpad3d
|
||
RePU
|
||
resistance_distance_exp
|
||
resize_nearest
|
||
rgb_to_bayer
|
||
rgb_to_grayscale
|
||
rgb_to_xyz
|
||
rgb_to_yuv
|
||
RMAF
|
||
rmsnorm
|
||
rmsnorm_residual
|
||
rmsnorm_silu
|
||
robustscalegate
|
||
robustscalehuber
|
||
rogers_tanimoto_silu
|
||
roiaware_pool1d
|
||
roll2d
|
||
roll3d
|
||
rope
|
||
SAAF
|
||
SaRa
|
||
SbPiPLU
|
||
scalenorm
|
||
scatter_add
|
||
scatter_div
|
||
scatter_nd
|
||
sceloss
|
||
SCLMish
|
||
SCSwish
|
||
segment_reduce
|
||
SELU+Affine
|
||
SELU-Affine-Gate
|
||
selu_clip
|
||
Sep
|
||
separable_conv2d
|
||
separable_conv3d
|
||
serf
|
||
Serlu
|
||
ShiftedSincUnit
|
||
Sigmoid+Clamp+Affine
|
||
Sigmoid-Affine-Gate
|
||
sigmoid_derivative
|
||
sigmoid_focal_loss
|
||
SigmoidGLU
|
||
SigmoidSlope-Affine-Gate
|
||
SigmoidSquared-Affine-Gate
|
||
signmuladd
|
||
SimCLRLoss
|
||
Sin-Affine-Gate
|
||
Sinc-Affine-Gate
|
||
sincoshypot
|
||
Sine-Affine-Gate
|
||
Sinh-Affine-Gate
|
||
SinLU
|
||
SinuGaussian
|
||
smelu
|
||
SmoothAbs
|
||
SmoothL1Loss
|
||
SmoothMaximum
|
||
SmoothMinimum
|
||
SmoothRamp
|
||
SmoothStep
|
||
SoftClip
|
||
SoftClip+Affine
|
||
SoftExponential
|
||
softmarginloss
|
||
softmax
|
||
softmin
|
||
softplus
|
||
Softplus+Bias
|
||
Softplus^2-Affine-Gate
|
||
Softplus^3-Affine-Gate
|
||
SoftplusGLU
|
||
SoftplusSqrt-Affine-Gate
|
||
softshrink
|
||
SoftShrink-Affine-Gate
|
||
softsign
|
||
Softsign+Affine
|
||
SoftSign-Affine-Gate
|
||
Softsign-Sigmoid-Mix-Gate
|
||
SoftSignSquared-Affine-Gate
|
||
solarize
|
||
Spatial-Diff Sigmoid Gate
|
||
spherefaceloss
|
||
Spherical+Affine
|
||
SQRBF
|
||
Sqrt(ReLU)+Affine
|
||
sqrtreciprocalrsqrt
|
||
Square-Sigmoid-Affine-Gate
|
||
squared_euclidean
|
||
SquaredHingeLoss
|
||
squareplus
|
||
SReLU
|
||
SRS
|
||
std_mean
|
||
STL
|
||
structural_similarity_softplus
|
||
SupConLoss
|
||
SwAT
|
||
swiglu
|
||
swish
|
||
Swish(SiLU)-Affine-Gate
|
||
Swish^2+Bias
|
||
swish_layernorm
|
||
switchablenorm
|
||
SørensenDice
|
||
Tanh-Affine-Gate
|
||
tanhexp
|
||
TanhGLU
|
||
TanhLU
|
||
tanhshrink
|
||
Tanhshrink+Affine
|
||
TanhSlope-Affine-Gate
|
||
TanhSoft1
|
||
TanhSoft2
|
||
TanimotoCoefficient
|
||
TDLoss
|
||
TeLU
|
||
Temperature Softmax Channel Affine
|
||
tensor_roll
|
||
three_interpolate
|
||
thresholdscalenegate
|
||
topk_filtering
|
||
total_correlation_elu
|
||
transpose_scale
|
||
tripletloss
|
||
tripletmarginloss
|
||
TripletMarginWithDistanceLoss
|
||
TrustRegionPolicyOptimizationLoss
|
||
TsallisDivergenceLoss
|
||
tversky_loss
|
||
TverskyIndex
|
||
tverskyloss
|
||
TweedieLoss
|
||
upsample
|
||
ValueLoss
|
||
Variance
|
||
VariationOfInformation
|
||
vecdot
|
||
VIDLoss
|
||
voxel_hash
|
||
voxel_mean
|
||
voxel_to_point
|
||
wasserstein_energy_gelu
|
||
wasserstein_layernorm
|
||
waveletaffine
|
||
WeightDecay
|
||
wingloss
|
||
winsorize_scale_normalize
|
||
wiou
|
||
xIELU
|
||
xSiLU
|
||
xyz_to_rgb
|
||
yuv_to_rgb
|
||
zeropad2d
|
||
zeropad3d
|
||
zscoresigmoiddenormalize
|
||
仿射+ReLU
|