PyPI - mindspore - Versions diffs - 2.6.0__cp39-none-any.whl → 2.7.0__cp39-none-any.whl - Mend - Supply Chain Defender

mindspore 2.6.0cp39-none-any.whl → 2.7.0cp39-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of mindspore might be problematic. Click here for more details.

Files changed (4804) hide show

mindspore/include/mindspore/ccsrc/pynative/grad/function/auto_generate/pyboost_native_grad_functions.h CHANGED Viewed

@@ -19,516 +19,547 @@ class NativeFunc {
     static NodePtr RunOpInVm(const PrimitivePtr &prim, const NodePtrList &inputs);
     static NodePtr RunOpDeprecated(const PrimitivePtr &prim, const NodePtrList &inputs);
     static ValuePtr ConvertNode2Value(const NodePtr &node);
-    static NodePtr Sinc(const NodePtr &input);
-    static NodePtr ReLU(const NodePtr &input);
-    static NodePtr L1LossBackwardExt(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
-    static NodePtr Log(const NodePtr &input);
-    static NodePtr AvgPool1D(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad);
-    static NodePtr BroadcastToView(const NodePtr &input, const NodePtr &shape);
-    static NodePtr GroupNormGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &mean, const NodePtr &rstd, const NodePtr &gamma_opt, const NodePtr &num_groups, const NodePtr &dx_is_require, const NodePtr &dgamma_is_require, const NodePtr &dbeta_is_require);
-    static NodePtr GridSampler2DGrad(const NodePtr &grad, const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners, const NodePtr &output_mask);
-    static NodePtr Conv3DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr Repeat(const NodePtr &input, const NodePtr &repeats);
-    static NodePtr Maximum(const NodePtr &input, const NodePtr &other);
-    static NodePtr InplaceSubScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr ScatterAddExt(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
-    static NodePtr TraceExt(const NodePtr &input);
-    static NodePtr InplaceMaskedFillScalar(const NodePtr &input, const NodePtr &mask, const NodePtr &value);
-    static NodePtr Select(const NodePtr &condition, const NodePtr &input, const NodePtr &other);
+    static NodePtr Cummax(const NodePtr &input, const NodePtr &axis);
+    static NodePtr ExpandAs(const NodePtr &input, const NodePtr &other);
+    static NodePtr InplaceErfinv(const NodePtr &input);
+    static NodePtr EluGradExt(const NodePtr &dout, const NodePtr &x_or_out, const NodePtr &alpha, const NodePtr &is_result);
+    static NodePtr InplaceDivs(const NodePtr &input, const NodePtr &other);
     static NodePtr GreaterEqual(const NodePtr &input, const NodePtr &other);
-    static NodePtr ConstantPadND(const NodePtr &input, const NodePtr &padding, const NodePtr &value);
-    static NodePtr ReflectionPad2DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
-    static NodePtr AsinExt(const NodePtr &input);
-    static NodePtr HardtanhGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
-    static NodePtr InnerCommAllGather(const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
-    static NodePtr ScatterValue(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
-    static NodePtr LogSoftmax(const NodePtr &logits, const NodePtr &axis);
+    static NodePtr InplaceUniform(const NodePtr &input, const NodePtr &from_, const NodePtr &to, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr ArgMinExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr Arange(const NodePtr &start, const NodePtr &end, const NodePtr &step, const NodePtr &dtype);
     static NodePtr AvgPool3DGradExt(const NodePtr &grad, const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
+    static NodePtr Sqrt(const NodePtr &x);
+    static NodePtr SmoothL1LossGrad(const NodePtr &prediction, const NodePtr &target, const NodePtr &dout, const NodePtr &beta, const NodePtr &reduction);
+    static NodePtr ScatterValue(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
     static NodePtr ReflectionPad3DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
-    static NodePtr NLLLoss2dGrad(const NodePtr &loss_grad, const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index, const NodePtr &total_weight);
-    static NodePtr InplaceFloor(const NodePtr &input);
-    static NodePtr MaxDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr AtanExt(const NodePtr &input);
-    static NodePtr FlashAttentionScoreGrad(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &dy, const NodePtr &pse_shift, const NodePtr &drop_mask, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &softmax_max, const NodePtr &softmax_sum, const NodePtr &softmax_in, const NodePtr &attention_in, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &head_num, const NodePtr &keep_prob, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &input_layout, const NodePtr &sparse_mode);
-    static NodePtr BatchNormGatherStatsWithCounts(const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &running_mean, const NodePtr &running_var, const NodePtr &momentum, const NodePtr &eps, const NodePtr &counts);
-    static NodePtr MaskedSelect(const NodePtr &input, const NodePtr &mask);
-    static NodePtr MoeTokenUnpermute(const NodePtr &permuted_tokens, const NodePtr &sorted_indices, const NodePtr &probs, const NodePtr &padded_mode, const NodePtr &restore_shape);
-    static NodePtr SwigluGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &dim);
-    static NodePtr BitwiseAndScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr Randn(const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr ConvTranspose2D(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &dilation);
-    static NodePtr CountNonZero(const NodePtr &input, const NodePtr &dim);
+    static NodePtr InplaceIndexPut(const NodePtr &input, const NodePtr &indices, const NodePtr &values, const NodePtr &accumulate);
+    static NodePtr DistCommReduceScatter(const NodePtr &other, const NodePtr &input_list, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr Abs(const NodePtr &input);
+    static NodePtr NLLLoss(const NodePtr &logits, const NodePtr &labels, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index);
+    static NodePtr Exp2(const NodePtr &input);
+    static NodePtr Muls(const NodePtr &input, const NodePtr &other);
+    static NodePtr LinSpaceExt(const NodePtr &start, const NodePtr &end, const NodePtr &steps, const NodePtr &dtype);
+    static NodePtr MaxPoolWithIndices(const NodePtr &x, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
+    static NodePtr L1LossExt(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
+    static NodePtr Add(const NodePtr &input, const NodePtr &other);
+    static NodePtr LinalgQr(const NodePtr &A, const NodePtr &mode);
+    static NodePtr InplaceAddExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr BroadcastToView(const NodePtr &input, const NodePtr &shape);
+    static NodePtr Expm1(const NodePtr &input);
+    static NodePtr SiLUGrad(const NodePtr &dout, const NodePtr &x);
+    static NodePtr IncreFlashAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &attn_mask, const NodePtr &actual_seq_lengths, const NodePtr &pse_shift, const NodePtr &dequant_scale1, const NodePtr &quant_scale1, const NodePtr &dequant_scale2, const NodePtr &quant_scale2, const NodePtr &quant_offset2, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &block_table, const NodePtr &kv_padding_size, const NodePtr &num_heads, const NodePtr &input_layout, const NodePtr &scale_value, const NodePtr &num_key_value_heads, const NodePtr &block_size, const NodePtr &inner_precise);
+    static NodePtr ConvolutionStrGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &output_mask);
+    static NodePtr NewFull(const NodePtr &input, const NodePtr &size, const NodePtr &fill_value, const NodePtr &dtype);
+    static NodePtr HShrinkGrad(const NodePtr &gradients, const NodePtr &features, const NodePtr &lambd);
+    static NodePtr NLLLoss2d(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index);
+    static NodePtr Embedding(const NodePtr &input, const NodePtr &weight, const NodePtr &padding_idx, const NodePtr &max_norm, const NodePtr &norm_type, const NodePtr &scale_grad_by_freq);
+    static NodePtr SubScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr Copy(const NodePtr &input);
+    static NodePtr InplaceDiv(const NodePtr &input, const NodePtr &other);
+    static NodePtr Swiglu(const NodePtr &input, const NodePtr &dim);
     static NodePtr FullLike(const NodePtr &input, const NodePtr &fill_value, const NodePtr &dtype);
-    static NodePtr FlashAttentionScore(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &real_shift, const NodePtr &drop_mask, const NodePtr &padding_mask, const NodePtr &attn_mask, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &head_num, const NodePtr &keep_prob, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &input_layout, const NodePtr &sparse_mode);
-    static NodePtr RandLikeExt(const NodePtr &tensor, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr TriangularSolve(const NodePtr &b, const NodePtr &A, const NodePtr &upper, const NodePtr &transpose, const NodePtr &unitriangular);
-    static NodePtr EluGradExt(const NodePtr &dout, const NodePtr &x_or_out, const NodePtr &alpha, const NodePtr &is_result);
-    static NodePtr NarrowView(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &length);
-    static NodePtr GridSampler3D(const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners);
-    static NodePtr MatrixInverseExt(const NodePtr &input);
-    static NodePtr SigmoidGrad(const NodePtr &y, const NodePtr &dy);
-    static NodePtr Addbmm(const NodePtr &input, const NodePtr &batch1, const NodePtr &batch2, const NodePtr &beta, const NodePtr &alpha);
-    static NodePtr Max(const NodePtr &input);
-    static NodePtr Exp(const NodePtr &input);
-    static NodePtr ReflectionPad1DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
+    static NodePtr Conv2DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr RemainderScalarTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr LogSigmoidGrad(const NodePtr &dy, const NodePtr &input, const NodePtr &buffer);
+    static NodePtr Addmm(const NodePtr &input, const NodePtr &mat1, const NodePtr &mat2, const NodePtr &beta, const NodePtr &alpha);
+    static NodePtr BatchNormExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &running_mean, const NodePtr &runnning_var, const NodePtr &training, const NodePtr &momentum, const NodePtr &epsilon);
+    static NodePtr UniformExt(const NodePtr &tensor, const NodePtr &a, const NodePtr &b, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr Erfc(const NodePtr &input);
+    static NodePtr ArgMaxWithValue(const NodePtr &input, const NodePtr &axis, const NodePtr &keep_dims);
+    static NodePtr ReplicationPad3D(const NodePtr &input, const NodePtr &padding);
     static NodePtr Unique2(const NodePtr &input, const NodePtr &sorted, const NodePtr &return_inverse, const NodePtr &return_counts);
-    static NodePtr InplaceThreshold(const NodePtr &input, const NodePtr &threshold, const NodePtr &value);
+    static NodePtr RemainderTensorTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr BitwiseXorScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr SumExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr MultinomialExt(const NodePtr &input, const NodePtr &num_samples, const NodePtr &replacement, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr DistCommAllGather(const NodePtr &tensor_list, const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr AdaptiveMaxPool1D(const NodePtr &input, const NodePtr &output_size);
+    static NodePtr KLDivGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &reduction, const NodePtr &log_target);
+    static NodePtr InplaceAddsExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr SoftmaxBackward(const NodePtr &dout, const NodePtr &out, const NodePtr &dim);
+    static NodePtr DistCommGather(const NodePtr &input, const NodePtr &gather_list, const NodePtr &rank_size, const NodePtr &dst, const NodePtr &rank_id, const NodePtr &group);
     static NodePtr Slice(const NodePtr &input, const NodePtr &begin, const NodePtr &size);
-    static NodePtr UpsampleLinear1D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr SliceExt(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &end, const NodePtr &step);
-    static NodePtr ReflectionPad2D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr MoeTokenPermuteGrad(const NodePtr &permuted_tokens_grad, const NodePtr &sorted_indices, const NodePtr &num_topk, const NodePtr &padded_mode);
-    static NodePtr InnerCommReduceScatter(const NodePtr &input, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
-    static NodePtr Minimum(const NodePtr &input, const NodePtr &other);
-    static NodePtr Mm(const NodePtr &input, const NodePtr &mat2);
-    static NodePtr BinaryCrossEntropyGrad(const NodePtr &input, const NodePtr &target, const NodePtr &grad_output, const NodePtr &weight, const NodePtr &reduction);
     static NodePtr Contiguous(const NodePtr &input);
-    static NodePtr SelectExtView(const NodePtr &input, const NodePtr &dim, const NodePtr &index);
-    static NodePtr Lerp(const NodePtr &input, const NodePtr &end, const NodePtr &weight);
-    static NodePtr Cross(const NodePtr &input, const NodePtr &other, const NodePtr &dim);
-    static NodePtr SortExt(const NodePtr &input, const NodePtr &dim, const NodePtr &descending, const NodePtr &stable);
-    static NodePtr Meshgrid(const NodePtr &inputs, const NodePtr &indexing);
-    static NodePtr BincountExt(const NodePtr &input, const NodePtr &weights, const NodePtr &minlength);
-    static NodePtr GluGrad(const NodePtr &grads, const NodePtr &x, const NodePtr &axis);
-    static NodePtr RandpermExt(const NodePtr &n, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr InplaceFloorDivides(const NodePtr &input, const NodePtr &other);
-    static NodePtr NLLLoss(const NodePtr &logits, const NodePtr &labels, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index);
-    static NodePtr AddLayerNormV2(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &beta, const NodePtr &epsilon, const NodePtr &additionalOut);
-    static NodePtr InplaceHardtanh(const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
-    static NodePtr AddcmulExt(const NodePtr &input, const NodePtr &tensor1, const NodePtr &tensor2, const NodePtr &value);
-    static NodePtr DistCommBatchIsendIrecv(const NodePtr &input, const NodePtr &group, const NodePtr &op_types, const NodePtr &remotes_ranks);
-    static NodePtr TopkExt(const NodePtr &input, const NodePtr &k, const NodePtr &dim, const NodePtr &largest, const NodePtr &sorted);
-    static NodePtr Addmm(const NodePtr &input, const NodePtr &mat1, const NodePtr &mat2, const NodePtr &beta, const NodePtr &alpha);
-    static NodePtr LayerNormExt(const NodePtr &input, const NodePtr &normalized_shape, const NodePtr &weight, const NodePtr &bias, const NodePtr &eps);
-    static NodePtr Var(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
-    static NodePtr Copy(const NodePtr &input);
-    static NodePtr ChunkView(const NodePtr &input, const NodePtr &chunks, const NodePtr &dim);
+    static NodePtr MaskedSelectGrad(const NodePtr &input, const NodePtr &mask, const NodePtr &grad);
+    static NodePtr ConstantPadND(const NodePtr &input, const NodePtr &padding, const NodePtr &value);
+    static NodePtr InplaceFillDiagonal(const NodePtr &input, const NodePtr &fill_value, const NodePtr &wrap);
+    static NodePtr Neg(const NodePtr &input);
+    static NodePtr InplaceBernoulliTensor(const NodePtr &input, const NodePtr &p, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr NewOnes(const NodePtr &input, const NodePtr &size, const NodePtr &dtype);
     static NodePtr InplaceTanh(const NodePtr &input);
-    static NodePtr InplaceMuls(const NodePtr &input, const NodePtr &other);
-    static NodePtr IsNegInf(const NodePtr &input);
-    static NodePtr LinalgQr(const NodePtr &A, const NodePtr &mode);
-    static NodePtr InnerInplaceIndexPut(const NodePtr &input, const NodePtr &indices, const NodePtr &values, const NodePtr &accumulate);
-    static NodePtr Min(const NodePtr &input);
+    static NodePtr Ceil(const NodePtr &input);
+    static NodePtr CrossEntropyLossGrad(const NodePtr &grad_loss, const NodePtr &log_prob, const NodePtr &target, const NodePtr &weight, const NodePtr &grad_zloss, const NodePtr &lse_for_zloss, const NodePtr &reduction, const NodePtr &ignore_index, const NodePtr &label_smoothing, const NodePtr &lse_square_scale_for_zloss);
+    static NodePtr ReflectionPad2DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
+    static NodePtr InplaceDivMods(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
+    static NodePtr Conv1DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr LogicalOr(const NodePtr &x, const NodePtr &y);
+    static NodePtr SplitTensorView(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
+    static NodePtr SearchSorted(const NodePtr &sorted_sequence, const NodePtr &values, const NodePtr &sorter, const NodePtr &dtype, const NodePtr &right);
+    static NodePtr DistCommReduceScatterTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr OneHotExt(const NodePtr &tensor, const NodePtr &num_classes, const NodePtr &on_value, const NodePtr &off_value, const NodePtr &axis);
+    static NodePtr UpsampleTrilinear3D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr InplaceCopy(const NodePtr &input, const NodePtr &src, const NodePtr &non_blocking);
+    static NodePtr Maximum(const NodePtr &input, const NodePtr &other);
     static NodePtr ExpandDims(const NodePtr &input_x, const NodePtr &axis);
-    static NodePtr IsClose(const NodePtr &input, const NodePtr &other, const NodePtr &rtol, const NodePtr &atol, const NodePtr &equal_nan);
-    static NodePtr LogicalNot(const NodePtr &input);
-    static NodePtr Log2(const NodePtr &input);
+    static NodePtr TransposeView(const NodePtr &input, const NodePtr &input_perm);
+    static NodePtr DistCommScatterTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &src, const NodePtr &rank_id, const NodePtr &group);
+    static NodePtr Select(const NodePtr &condition, const NodePtr &input, const NodePtr &other);
+    static NodePtr Squeeze(const NodePtr &input, const NodePtr &axis);
+    static NodePtr Dense(const NodePtr &input, const NodePtr &weight, const NodePtr &bias);
+    static NodePtr DropoutExt(const NodePtr &input, const NodePtr &p, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr Less(const NodePtr &input, const NodePtr &other);
+    static NodePtr AcosExt(const NodePtr &input);
+    static NodePtr PReLU(const NodePtr &input, const NodePtr &weight);
+    static NodePtr BatchNormReduceGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &weight, const NodePtr &input_g, const NodePtr &weight_g, const NodePtr &bias_g);
+    static NodePtr DiagExt(const NodePtr &input, const NodePtr &diagonal);
+    static NodePtr LerpScalar(const NodePtr &input, const NodePtr &end, const NodePtr &weight);
+    static NodePtr Lerp(const NodePtr &input, const NodePtr &end, const NodePtr &weight);
+    static NodePtr ConvTranspose2D(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &dilation);
+    static NodePtr InplaceScatterAdd(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
+    static NodePtr FlashAttentionScoreGrad(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &dy, const NodePtr &pse_shift, const NodePtr &drop_mask, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &softmax_max, const NodePtr &softmax_sum, const NodePtr &softmax_in, const NodePtr &attention_in, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &head_num, const NodePtr &keep_prob, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &input_layout, const NodePtr &sparse_mode);
+    static NodePtr ReflectionPad1DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
     static NodePtr InplaceIndexAddExt(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &source, const NodePtr &alpha);
+    static NodePtr MaxPoolGradWithIndices(const NodePtr &x, const NodePtr &grad, const NodePtr &argmax, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
+    static NodePtr InplaceFloorDivide(const NodePtr &input, const NodePtr &other);
+    static NodePtr InplaceMuls(const NodePtr &input, const NodePtr &other);
+    static NodePtr ReplicationPad1DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
+    static NodePtr Repeat(const NodePtr &input, const NodePtr &repeats);
+    static NodePtr RemainderTensorScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr InnerInplaceIndexPut(const NodePtr &input, const NodePtr &indices, const NodePtr &values, const NodePtr &accumulate);
+    static NodePtr ClampScalar(const NodePtr &input, const NodePtr &min, const NodePtr &max);
+    static NodePtr HSwish(const NodePtr &input);
+    static NodePtr NormalFloatTensor(const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr BitwiseOrScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr OnesLikeExt(const NodePtr &input, const NodePtr &dtype);
     static NodePtr HShrink(const NodePtr &input, const NodePtr &lambd);
-    static NodePtr DistCommReduceScatter(const NodePtr &other, const NodePtr &input_list, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr LogSoftmaxExt(const NodePtr &input, const NodePtr &dim, const NodePtr &dtype);
+    static NodePtr IsClose(const NodePtr &input, const NodePtr &other, const NodePtr &rtol, const NodePtr &atol, const NodePtr &equal_nan);
+    static NodePtr FlashAttentionScore(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &real_shift, const NodePtr &drop_mask, const NodePtr &padding_mask, const NodePtr &attn_mask, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &head_num, const NodePtr &keep_prob, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &input_layout, const NodePtr &sparse_mode);
+    static NodePtr AddcmulExt(const NodePtr &input, const NodePtr &tensor1, const NodePtr &tensor2, const NodePtr &value);
+    static NodePtr Equal(const NodePtr &input, const NodePtr &other);
+    static NodePtr CumsumExt(const NodePtr &input, const NodePtr &dim, const NodePtr &dtype);
+    static NodePtr NLLLossGrad(const NodePtr &logits, const NodePtr &loss_grad, const NodePtr &labels, const NodePtr &weight, const NodePtr &total_weight, const NodePtr &reduction, const NodePtr &ignore_index);
+    static NodePtr Outer(const NodePtr &input, const NodePtr &vec2);
+    static NodePtr Sub(const NodePtr &input, const NodePtr &other);
+    static NodePtr MatMulExt(const NodePtr &input, const NodePtr &other);
     static NodePtr InplaceZero(const NodePtr &input);
+    static NodePtr ReshapeAndCache(const NodePtr &key, const NodePtr &value, const NodePtr &key_cache, const NodePtr &value_cache, const NodePtr &slot_mapping);
+    static NodePtr RandIntLike(const NodePtr &input, const NodePtr &low, const NodePtr &high, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr AsinExt(const NodePtr &input);
+    static NodePtr InnerNonZero(const NodePtr &input);
+    static NodePtr RandnLike(const NodePtr &input, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr NeScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr HSwishGrad(const NodePtr &y_grad, const NodePtr &x);
+    static NodePtr BitwiseOrTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr Pow(const NodePtr &input, const NodePtr &exponent);
+    static NodePtr ReplicationPad1D(const NodePtr &input, const NodePtr &padding);
+    static NodePtr DistCommAllToAllV(const NodePtr &other, const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size);
+    static NodePtr Baddbmm(const NodePtr &input, const NodePtr &batch1, const NodePtr &batch2, const NodePtr &beta, const NodePtr &alpha);
+    static NodePtr InnerIndex(const NodePtr &input, const NodePtr &indices);
     static NodePtr AsinhExt(const NodePtr &input);
-    static NodePtr ReplicationPad1DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
-    static NodePtr ReplicationPad3DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
-    static NodePtr Conv1DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr BatchNormGradExt(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &running_mean, const NodePtr &running_var, const NodePtr &saved_mean, const NodePtr &saved_rstd, const NodePtr &training, const NodePtr &eps, const NodePtr &output_mask);
-    static NodePtr AdaptiveAvgPool3DGradExt(const NodePtr &input_grad, const NodePtr &input);
-    static NodePtr InplaceFillTensor(const NodePtr &input, const NodePtr &value);
-    static NodePtr Concat(const NodePtr &tensors, const NodePtr &axis);
-    static NodePtr Less(const NodePtr &input, const NodePtr &other);
-    static NodePtr BitwiseXorTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr DistCommAllReduce(const NodePtr &input, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr AdamW(const NodePtr &var, const NodePtr &m, const NodePtr &v, const NodePtr &max_v, const NodePtr &gradient, const NodePtr &step, const NodePtr &lr, const NodePtr &beta1, const NodePtr &beta2, const NodePtr &decay, const NodePtr &eps, const NodePtr &amsgrad, const NodePtr &maximize);
+    static NodePtr DistCommBatchIsendIrecv(const NodePtr &input, const NodePtr &group, const NodePtr &op_types, const NodePtr &remotes_ranks);
+    static NodePtr ExpandDimsView(const NodePtr &input, const NodePtr &dim);
+    static NodePtr GatherD(const NodePtr &x, const NodePtr &dim, const NodePtr &index);
+    static NodePtr Generator(const NodePtr &cmd, const NodePtr &inputs);
+    static NodePtr Atanh(const NodePtr &input);
     static NodePtr RmsNorm(const NodePtr &x, const NodePtr &gamma, const NodePtr &epsilon);
-    static NodePtr IndexFillScalar(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value);
-    static NodePtr HSwishGrad(const NodePtr &y_grad, const NodePtr &x);
-    static NodePtr Addmv(const NodePtr &input, const NodePtr &mat, const NodePtr &vec, const NodePtr &beta, const NodePtr &alpha);
-    static NodePtr UniqueDim(const NodePtr &input, const NodePtr &sorted, const NodePtr &return_inverse, const NodePtr &dim);
-    static NodePtr SliceExtView(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &end, const NodePtr &step);
-    static NodePtr InplaceStopGradient(const NodePtr &input);
-    static NodePtr StackExt(const NodePtr &tensors, const NodePtr &dim);
-    static NodePtr InplaceRandom(const NodePtr &input, const NodePtr &from_, const NodePtr &to, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr MatMul(const NodePtr &input, const NodePtr &mat2, const NodePtr &transpose_a, const NodePtr &transpose_b);
-    static NodePtr SoftplusGradExt(const NodePtr &dout, const NodePtr &x, const NodePtr &beta, const NodePtr &threshold);
-    static NodePtr ViewAs(const NodePtr &input, const NodePtr &other);
-    static NodePtr Add(const NodePtr &input, const NodePtr &other);
-    static NodePtr DistCommBarrier(const NodePtr &group);
-    static NodePtr ConvolutionStr(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups);
+    static NodePtr Kthvalue(const NodePtr &input, const NodePtr &k, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr Sign(const NodePtr &input);
     static NodePtr AllGatherMatmul(const NodePtr &input, const NodePtr &x2, const NodePtr &group, const NodePtr &world_size, const NodePtr &bias, const NodePtr &gather_index, const NodePtr &gather_output, const NodePtr &comm_turn, const NodePtr &trans_input, const NodePtr &trans_x2);
-    static NodePtr Reciprocal(const NodePtr &input);
-    static NodePtr InplaceDiv(const NodePtr &input, const NodePtr &other);
+    static NodePtr GroupNorm(const NodePtr &input, const NodePtr &num_groups, const NodePtr &weight, const NodePtr &bias, const NodePtr &eps);
+    static NodePtr LogAddExp(const NodePtr &input, const NodePtr &other);
+    static NodePtr HardtanhGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
+    static NodePtr InplaceMaskedFillTensor(const NodePtr &input, const NodePtr &mask, const NodePtr &value);
+    static NodePtr InplaceScatterSrc(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
+    static NodePtr AvgPool3DExt(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
+    static NodePtr InplaceDivMod(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
+    static NodePtr Log10(const NodePtr &input);
+    static NodePtr InplaceSiLU(const NodePtr &input);
+    static NodePtr Mv(const NodePtr &input, const NodePtr &vec);
+    static NodePtr NormalTensorTensor(const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr AvgPool2D(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
+    static NodePtr UpsampleNearest2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr MedianExt(const NodePtr &input);
+    static NodePtr RotaryPositionEmbedding(const NodePtr &x, const NodePtr &cos, const NodePtr &sin, const NodePtr &mode);
     static NodePtr FlattenExt(const NodePtr &input, const NodePtr &start_dim, const NodePtr &end_dim);
-    static NodePtr InplacePut(const NodePtr &input, const NodePtr &index, const NodePtr &source, const NodePtr &accumulate);
-    static NodePtr BinaryCrossEntropy(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction);
-    static NodePtr InplaceScatterSrcReduce(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
+    static NodePtr AdaptiveAvgPool1D(const NodePtr &input, const NodePtr &output_size);
+    static NodePtr Conv3DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr BatchNormStats(const NodePtr &input, const NodePtr &eps);
+    static NodePtr MaxPoolWithMask(const NodePtr &x, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
+    static NodePtr ReLU(const NodePtr &input);
+    static NodePtr MSELossGradExt(const NodePtr &dout, const NodePtr &x, const NodePtr &target, const NodePtr &reduction);
     static NodePtr Zeros(const NodePtr &size, const NodePtr &dtype);
-    static NodePtr RepeatInterleaveGrad(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim);
-    static NodePtr AdaptiveAvgPool2DGradExt(const NodePtr &grad_output, const NodePtr &x);
-    static NodePtr AsStrided(const NodePtr &input, const NodePtr &size, const NodePtr &stride, const NodePtr &storage_offset);
-    static NodePtr NewZeros(const NodePtr &input, const NodePtr &size, const NodePtr &dtype);
-    static NodePtr ClampTensor(const NodePtr &input, const NodePtr &min, const NodePtr &max);
-    static NodePtr Muls(const NodePtr &input, const NodePtr &other);
-    static NodePtr Tan(const NodePtr &input);
-    static NodePtr Conv2DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr InplaceAddmm(const NodePtr &input, const NodePtr &mat1, const NodePtr &mat2, const NodePtr &beta, const NodePtr &alpha);
-    static NodePtr UpsampleNearest2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr Frac(const NodePtr &input);
-    static NodePtr Generator(const NodePtr &cmd, const NodePtr &inputs);
-    static NodePtr RepeatInterleaveInt(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim, const NodePtr &output_size);
-    static NodePtr VarMean(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
-    static NodePtr Ceil(const NodePtr &input);
-    static NodePtr InplaceReLU(const NodePtr &input);
-    static NodePtr MishExt(const NodePtr &input);
+    static NodePtr BernoulliExt(const NodePtr &input, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr Triu(const NodePtr &input, const NodePtr &diagonal);
+    static NodePtr Mla(const NodePtr &query, const NodePtr &q_rope, const NodePtr &kv_cache, const NodePtr &k_rope, const NodePtr &block_tables, const NodePtr &attn_mask, const NodePtr &deq_scale_qk, const NodePtr &deq_scale_pv, const NodePtr &q_seq_lens, const NodePtr &context_lens, const NodePtr &head_num, const NodePtr &scale_value, const NodePtr &kv_head_num, const NodePtr &mask_mode, const NodePtr &is_ring);
+    static NodePtr DistCommAllToAllVSingle(const NodePtr &other, const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size, const NodePtr &split_sizes_empty);
+    static NodePtr UpsampleNearest2D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr RandpermExt(const NodePtr &n, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr SpeedFusionAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &head_num, const NodePtr &input_layout, const NodePtr &seed, const NodePtr &offset, const NodePtr &pse, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &scale, const NodePtr &keep_prob, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &sparse_mode, const NodePtr &gen_mask_parallel, const NodePtr &sync, const NodePtr &pse_type, const NodePtr &q_start_idx, const NodePtr &kv_start_idx);
+    static NodePtr NonZeroExt(const NodePtr &input);
+    static NodePtr SliceExtView(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &end, const NodePtr &step);
+    static NodePtr EluExt(const NodePtr &input, const NodePtr &alpha);
+    static NodePtr MaskedSelect(const NodePtr &input, const NodePtr &mask);
+    static NodePtr GridSampler2DGrad(const NodePtr &grad, const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners, const NodePtr &output_mask);
+    static NodePtr AddScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr Conv1DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr ReluGrad(const NodePtr &y_backprop, const NodePtr &x);
+    static NodePtr RandLikeExt(const NodePtr &tensor, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr BatchMatMul(const NodePtr &x, const NodePtr &y, const NodePtr &transpose_a, const NodePtr &transpose_b);
+    static NodePtr DistCommGatherIntoTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &dst, const NodePtr &rank_id, const NodePtr &group);
+    static NodePtr LogSoftmax(const NodePtr &logits, const NodePtr &axis);
+    static NodePtr SiLU(const NodePtr &input);
+    static NodePtr MaxPoolGradWithMask(const NodePtr &x, const NodePtr &grad, const NodePtr &mask, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
+    static NodePtr Max(const NodePtr &input);
     static NodePtr GridSampler3DGrad(const NodePtr &grad, const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners, const NodePtr &output_mask);
-    static NodePtr InplaceErfinv(const NodePtr &input);
-    static NodePtr MaxPoolWithMask(const NodePtr &x, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
-    static NodePtr Sin(const NodePtr &input);
+    static NodePtr MoeTokenPermuteGrad(const NodePtr &permuted_tokens_grad, const NodePtr &sorted_indices, const NodePtr &num_topk, const NodePtr &padded_mode);
+    static NodePtr ReflectionPad1D(const NodePtr &input, const NodePtr &padding);
+    static NodePtr MaxDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr Log(const NodePtr &input);
     static NodePtr ReduceAll(const NodePtr &input, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr InplaceFillScalar(const NodePtr &input, const NodePtr &value);
-    static NodePtr NotEqual(const NodePtr &input, const NodePtr &other);
-    static NodePtr NeScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr InnerCommIrecv(const NodePtr &tag, const NodePtr &src, const NodePtr &shape, const NodePtr &group, const NodePtr &dtype);
-    static NodePtr Convolution(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups);
-    static NodePtr ArgMinExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr BinaryCrossEntropyWithLogitsBackward(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &posWeight, const NodePtr &reduction);
-    static NodePtr Dense(const NodePtr &input, const NodePtr &weight, const NodePtr &bias);
-    static NodePtr BatchNormElemt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &mean, const NodePtr &invstd, const NodePtr &eps);
-    static NodePtr Arange(const NodePtr &start, const NodePtr &end, const NodePtr &step, const NodePtr &dtype);
-    static NodePtr SplitWithSize(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
-    static NodePtr SoftShrinkGrad(const NodePtr &input_grad, const NodePtr &input_x, const NodePtr &lambd);
-    static NodePtr DistCommAllGatherIntoTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr MaskedFill(const NodePtr &input_x, const NodePtr &mask, const NodePtr &value);
+    static NodePtr InplaceClampScalar(const NodePtr &input, const NodePtr &min, const NodePtr &max);
+    static NodePtr InnerMoeTokenUnpermute(const NodePtr &permuted_tokens, const NodePtr &sorted_indices, const NodePtr &probs, const NodePtr &padded_mode, const NodePtr &restore_shape);
+    static NodePtr ReduceMax(const NodePtr &x, const NodePtr &axis, const NodePtr &keep_dims);
+    static NodePtr ReflectionPad3D(const NodePtr &input, const NodePtr &padding);
+    static NodePtr MatmulReduceScatter(const NodePtr &input, const NodePtr &x2, const NodePtr &group, const NodePtr &world_size, const NodePtr &reduce_op, const NodePtr &bias, const NodePtr &comm_turn, const NodePtr &trans_input, const NodePtr &trans_x2);
+    static NodePtr ZerosLikeExt(const NodePtr &input, const NodePtr &dtype);
+    static NodePtr SoftShrink(const NodePtr &input, const NodePtr &lambd);
+    static NodePtr InplaceStopGradient(const NodePtr &input);
+    static NodePtr FloorDiv(const NodePtr &input, const NodePtr &other);
+    static NodePtr MoeDistributeDispatch(const NodePtr &x, const NodePtr &expert_ids, const NodePtr &ep_world_size, const NodePtr &ep_rank_id, const NodePtr &moe_expert_num, const NodePtr &expert_scales, const NodePtr &scales, const NodePtr &x_active_mask, const NodePtr &group_ep, const NodePtr &group_tp, const NodePtr &tp_world_size, const NodePtr &tp_rank_id, const NodePtr &expert_shard_type, const NodePtr &shared_expert_num, const NodePtr &shared_expert_rank_num, const NodePtr &quant_mode, const NodePtr &global_bs, const NodePtr &expert_token_nums_type);
     static NodePtr GLU(const NodePtr &x, const NodePtr &axis);
+    static NodePtr Exp(const NodePtr &input);
+    static NodePtr AcoshExt(const NodePtr &input);
+    static NodePtr ReflectionPad2D(const NodePtr &input, const NodePtr &padding);
+    static NodePtr SeLUExt(const NodePtr &input);
+    static NodePtr BroadcastTo(const NodePtr &input, const NodePtr &shape);
+    static NodePtr RepeatInterleaveInt(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim, const NodePtr &output_size);
     static NodePtr DropoutDoMaskExt(const NodePtr &input, const NodePtr &mask, const NodePtr &p);
-    static NodePtr MeanExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr Threshold(const NodePtr &input, const NodePtr &threshold, const NodePtr &value);
-    static NodePtr ReverseV2(const NodePtr &input, const NodePtr &axis);
-    static NodePtr Mul(const NodePtr &input, const NodePtr &other);
-    static NodePtr RepeatInterleaveTensor(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim, const NodePtr &output_size);
+    static NodePtr ReplicationPad2D(const NodePtr &input, const NodePtr &padding);
     static NodePtr ArgMinWithValue(const NodePtr &input, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr NonZero(const NodePtr &input);
-    static NodePtr SoftplusExt(const NodePtr &input, const NodePtr &beta, const NodePtr &threshold);
-    static NodePtr SoftMarginLossGrad(const NodePtr &predict, const NodePtr &label, const NodePtr &dout, const NodePtr &reduction);
-    static NodePtr SoftmaxBackward(const NodePtr &dout, const NodePtr &out, const NodePtr &dim);
-    static NodePtr ExpandAs(const NodePtr &input, const NodePtr &other);
-    static NodePtr HShrinkGrad(const NodePtr &gradients, const NodePtr &features, const NodePtr &lambd);
+    static NodePtr InplaceReLU(const NodePtr &input);
+    static NodePtr AddcdivExt(const NodePtr &input, const NodePtr &tensor1, const NodePtr &tensor2, const NodePtr &value);
+    static NodePtr GeLUGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &y);
+    static NodePtr EqualExt(const NodePtr &input, const NodePtr &other);
+    static NodePtr NotEqual(const NodePtr &input, const NodePtr &other);
+    static NodePtr MultiScaleDeformableAttnGrad(const NodePtr &value, const NodePtr &shape, const NodePtr &offset, const NodePtr &locations_trans, const NodePtr &weight, const NodePtr &grad_output);
+    static NodePtr InplaceSubScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr InplaceRemainderTensorScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr CountNonZero(const NodePtr &input, const NodePtr &dim);
+    static NodePtr MoeTokenUnpermuteGrad(const NodePtr &permuted_tokens, const NodePtr &unpermuted_tokens_grad, const NodePtr &sorted_indices, const NodePtr &probs, const NodePtr &padded_mode, const NodePtr &restore_shape);
+    static NodePtr MaskedScatter(const NodePtr &input, const NodePtr &mask, const NodePtr &source);
     static NodePtr FmodScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr PromptFlashAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &attn_mask, const NodePtr &actual_seq_lengths, const NodePtr &actual_seq_lengths_kv, const NodePtr &pse_shift, const NodePtr &deq_scale1, const NodePtr &quant_scale1, const NodePtr &deq_scale2, const NodePtr &quant_scale2, const NodePtr &quant_offset2, const NodePtr &num_heads, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &input_layout, const NodePtr &num_key_value_heads, const NodePtr &sparse_mode, const NodePtr &inner_precise);
-    static NodePtr Round(const NodePtr &input, const NodePtr &decimals);
-    static NodePtr RandExt(const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr BitwiseXorScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr LogSoftmaxGrad(const NodePtr &logits, const NodePtr &grad, const NodePtr &axis);
-    static NodePtr LogicalAnd(const NodePtr &x, const NodePtr &y);
-    static NodePtr TypeAs(const NodePtr &input, const NodePtr &other);
-    static NodePtr SubScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr Meshgrid(const NodePtr &inputs, const NodePtr &indexing);
+    static NodePtr SplitWithSizeView(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
+    static NodePtr ClampTensor(const NodePtr &input, const NodePtr &min, const NodePtr &max);
+    static NodePtr UpsampleNearest3D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr BitwiseAndScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr PowScalarTensor(const NodePtr &input, const NodePtr &exponent);
+    static NodePtr Clone(const NodePtr &input);
+    static NodePtr XLogYScalarOther(const NodePtr &input, const NodePtr &other);
+    static NodePtr Rsqrt(const NodePtr &input);
+    static NodePtr SmoothL1Loss(const NodePtr &prediction, const NodePtr &target, const NodePtr &beta, const NodePtr &reduction);
+    static NodePtr MSELossExt(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
+    static NodePtr TExt(const NodePtr &input);
+    static NodePtr IsNegInf(const NodePtr &input);
+    static NodePtr InplaceMul(const NodePtr &input, const NodePtr &other);
+    static NodePtr AdaptiveAvgPool3DGradExt(const NodePtr &input_grad, const NodePtr &input);
     static NodePtr BitwiseAndTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr FillTensor(const NodePtr &size, const NodePtr &fill_value, const NodePtr &dtype);
-    static NodePtr BatchNormElemtGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &weight, const NodePtr &sumd_dy, const NodePtr &sum_dy_xmu, const NodePtr &count);
-    static NodePtr UpsampleNearest1DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr Greater(const NodePtr &input, const NodePtr &other);
-    static NodePtr LogSoftmaxExt(const NodePtr &input, const NodePtr &dim, const NodePtr &dtype);
-    static NodePtr BatchNormExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &running_mean, const NodePtr &runnning_var, const NodePtr &training, const NodePtr &momentum, const NodePtr &epsilon);
+    static NodePtr Minimum(const NodePtr &input, const NodePtr &other);
+    static NodePtr SoftplusGradExt(const NodePtr &dout, const NodePtr &x, const NodePtr &beta, const NodePtr &threshold);
+    static NodePtr Softmax(const NodePtr &input, const NodePtr &axis);
+    static NodePtr Cast(const NodePtr &input, const NodePtr &dtype);
     static NodePtr TransposeExtView(const NodePtr &input, const NodePtr &dim0, const NodePtr &dim1);
-    static NodePtr RemainderTensorScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr BatchMatMulExt(const NodePtr &input, const NodePtr &mat2);
-    static NodePtr MultiScaleDeformableAttnGrad(const NodePtr &value, const NodePtr &shape, const NodePtr &offset, const NodePtr &locations_trans, const NodePtr &weight, const NodePtr &grad_output);
-    static NodePtr OnesLikeExt(const NodePtr &input, const NodePtr &dtype);
-    static NodePtr UpsampleNearest3DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr ClampScalar(const NodePtr &input, const NodePtr &min, const NodePtr &max);
-    static NodePtr Conv1DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr PReLUGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &weight);
-    static NodePtr SelectV2(const NodePtr &condition, const NodePtr &input, const NodePtr &other);
-    static NodePtr InplaceCopy(const NodePtr &input, const NodePtr &src);
-    static NodePtr InplaceClampTensor(const NodePtr &input, const NodePtr &min, const NodePtr &max);
-    static NodePtr ReplicationPad2D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr DistCommGather(const NodePtr &input, const NodePtr &gather_list, const NodePtr &rank_size, const NodePtr &dst, const NodePtr &rank_id, const NodePtr &group);
-    static NodePtr AdaptiveMaxPool1D(const NodePtr &input, const NodePtr &output_size);
-    static NodePtr LayerNormGradExt(const NodePtr &dy, const NodePtr &x, const NodePtr &normalized_shape, const NodePtr &mean, const NodePtr &variance, const NodePtr &gamma, const NodePtr &beta, const NodePtr &output_mask);
-    static NodePtr UpsampleNearest1D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr Split(const NodePtr &input_x, const NodePtr &axis, const NodePtr &output_num);
+    static NodePtr Randn(const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr NormalFloatFloat(const NodePtr &mean, const NodePtr &std, const NodePtr &size, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr BatchNormGradExt(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &running_mean, const NodePtr &running_var, const NodePtr &saved_mean, const NodePtr &saved_rstd, const NodePtr &training, const NodePtr &eps, const NodePtr &output_mask);
+    static NodePtr ApplyRotaryPosEmb(const NodePtr &query, const NodePtr &key, const NodePtr &cos, const NodePtr &sin, const NodePtr &position_ids, const NodePtr &cos_format);
+    static NodePtr InplaceFloorDivides(const NodePtr &input, const NodePtr &other);
+    static NodePtr Var(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
+    static NodePtr SortExt(const NodePtr &input, const NodePtr &dim, const NodePtr &descending, const NodePtr &stable);
+    static NodePtr InplaceFloor(const NodePtr &input);
+    static NodePtr InplaceNormal(const NodePtr &input, const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
     static NodePtr IsFinite(const NodePtr &input);
-    static NodePtr AdaptiveAvgPool1D(const NodePtr &input, const NodePtr &output_size);
-    static NodePtr InplaceScatterValueReduce(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value, const NodePtr &reduce);
-    static NodePtr StdMean(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
-    static NodePtr MaxPoolWithIndices(const NodePtr &x, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
-    static NodePtr Reshape(const NodePtr &input, const NodePtr &shape);
-    static NodePtr MaxPoolGradWithIndices(const NodePtr &x, const NodePtr &grad, const NodePtr &argmax, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
+    static NodePtr AsStrided(const NodePtr &input, const NodePtr &size, const NodePtr &stride, const NodePtr &storage_offset);
+    static NodePtr Floor(const NodePtr &input);
+    static NodePtr Transpose(const NodePtr &input, const NodePtr &input_perm);
+    static NodePtr SoftplusExt(const NodePtr &input, const NodePtr &beta, const NodePtr &threshold);
+    static NodePtr InplaceThreshold(const NodePtr &input, const NodePtr &threshold, const NodePtr &value);
+    static NodePtr IndexFillScalar(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value);
+    static NodePtr InnerCommAllGather(const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr NLLLoss2dGrad(const NodePtr &loss_grad, const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index, const NodePtr &total_weight);
+    static NodePtr LeakyReLUGradExt(const NodePtr &dy, const NodePtr &input, const NodePtr &negative_slope, const NodePtr &is_result);
+    static NodePtr InplacePut(const NodePtr &input, const NodePtr &index, const NodePtr &source, const NodePtr &accumulate);
+    static NodePtr InplaceLog(const NodePtr &input);
     static NodePtr AdaptiveMaxPool2D(const NodePtr &input, const NodePtr &output_size);
-    static NodePtr SoftShrink(const NodePtr &input, const NodePtr &lambd);
-    static NodePtr SiLU(const NodePtr &input);
-    static NodePtr InplaceScatterValue(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value);
-    static NodePtr UniformExt(const NodePtr &tensor, const NodePtr &a, const NodePtr &b, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr InplaceFloorDivide(const NodePtr &input, const NodePtr &other);
-    static NodePtr SeluGrad(const NodePtr &gradient, const NodePtr &result);
-    static NodePtr SpeedFusionAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &head_num, const NodePtr &input_layout, const NodePtr &seed, const NodePtr &offset, const NodePtr &pse, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &scale, const NodePtr &keep_prob, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &sparse_mode, const NodePtr &gen_mask_parallel, const NodePtr &sync, const NodePtr &pse_type, const NodePtr &q_start_idx, const NodePtr &kv_start_idx);
-    static NodePtr BitwiseOrScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr MoeTokenUnpermuteGrad(const NodePtr &permuted_tokens, const NodePtr &unpermuted_tokens_grad, const NodePtr &sorted_indices, const NodePtr &probs, const NodePtr &padded_mode, const NodePtr &restore_shape);
-    static NodePtr TExt(const NodePtr &input);
-    static NodePtr PowTensorScalar(const NodePtr &input, const NodePtr &exponent);
-    static NodePtr InnerCommAllToAllV(const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size, const NodePtr &split_sizes_empty);
-    static NodePtr Chunk(const NodePtr &input, const NodePtr &chunks, const NodePtr &dim);
-    static NodePtr Cosh(const NodePtr &input);
-    static NodePtr InplaceScatterSrc(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
-    static NodePtr SplitTensor(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
-    static NodePtr Baddbmm(const NodePtr &input, const NodePtr &batch1, const NodePtr &batch2, const NodePtr &beta, const NodePtr &alpha);
-    static NodePtr Sqrt(const NodePtr &x);
-    static NodePtr Exp2(const NodePtr &input);
-    static NodePtr GatherDGradV2(const NodePtr &x, const NodePtr &dim, const NodePtr &index, const NodePtr &dout);
-    static NodePtr Log1p(const NodePtr &input);
-    static NodePtr LogAddExp(const NodePtr &input, const NodePtr &other);
-    static NodePtr Log10(const NodePtr &input);
-    static NodePtr Std(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
-    static NodePtr Square(const NodePtr &input);
-    static NodePtr DistCommAllToAllV(const NodePtr &other, const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size);
-    static NodePtr FloorDivScalar(const NodePtr &input, const NodePtr &other);
-    static NodePtr BroadcastTo(const NodePtr &input, const NodePtr &shape);
-    static NodePtr FmodTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr InnerCommAllReduce(const NodePtr &input, const NodePtr &op_type, const NodePtr &group);
-    static NodePtr InplaceDivMod(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
-    static NodePtr InplaceNormal(const NodePtr &input, const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr UnstackExtView(const NodePtr &input, const NodePtr &dim);
-    static NodePtr AddExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr Cos(const NodePtr &input);
-    static NodePtr InplaceIndexPut(const NodePtr &input, const NodePtr &indices, const NodePtr &values, const NodePtr &accumulate);
-    static NodePtr ReflectionPad1D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr RandInt(const NodePtr &low, const NodePtr &high, const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr RemainderScalarTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr ReplicationPad1D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr Roll(const NodePtr &input, const NodePtr &shifts, const NodePtr &dims);
-    static NodePtr ReflectionPad3D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr Col2ImExt(const NodePtr &input, const NodePtr &output_size, const NodePtr &kernel_size, const NodePtr &dilation, const NodePtr &padding, const NodePtr &stride);
-    static NodePtr ReduceMax(const NodePtr &x, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr Nansum(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr L1LossExt(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
-    static NodePtr Hardtanh(const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
-    static NodePtr OneHotExt(const NodePtr &tensor, const NodePtr &num_classes, const NodePtr &on_value, const NodePtr &off_value, const NodePtr &axis);
     static NodePtr IsInf(const NodePtr &input);
-    static NodePtr AddcdivExt(const NodePtr &input, const NodePtr &tensor1, const NodePtr &tensor2, const NodePtr &value);
-    static NodePtr LogSigmoidGrad(const NodePtr &dy, const NodePtr &input, const NodePtr &buffer);
-    static NodePtr DropoutExt(const NodePtr &input, const NodePtr &p, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr InplaceDivs(const NodePtr &input, const NodePtr &other);
-    static NodePtr RandIntLike(const NodePtr &input, const NodePtr &low, const NodePtr &high, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr Index(const NodePtr &input, const NodePtr &indices);
-    static NodePtr GroupNorm(const NodePtr &input, const NodePtr &num_groups, const NodePtr &weight, const NodePtr &bias, const NodePtr &eps);
-    static NodePtr FloorDiv(const NodePtr &input, const NodePtr &other);
-    static NodePtr Neg(const NodePtr &input);
-    static NodePtr Divs(const NodePtr &input, const NodePtr &other);
-    static NodePtr PowScalarTensor(const NodePtr &input, const NodePtr &exponent);
-    static NodePtr LerpScalar(const NodePtr &input, const NodePtr &end, const NodePtr &weight);
-    static NodePtr LeakyReLUExt(const NodePtr &input, const NodePtr &negative_slope);
-    static NodePtr BatchNormReduceGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &weight, const NodePtr &input_g, const NodePtr &weight_g, const NodePtr &bias_g);
-    static NodePtr DistCommIrecv(const NodePtr &input, const NodePtr &tag, const NodePtr &src, const NodePtr &group);
-    static NodePtr InplaceElu(const NodePtr &input, const NodePtr &alpha);
-    static NodePtr FFNExt(const NodePtr &x, const NodePtr &weight1, const NodePtr &weight2, const NodePtr &expertTokens, const NodePtr &bias1, const NodePtr &bias2, const NodePtr &scale, const NodePtr &offset, const NodePtr &deqScale1, const NodePtr &deqScale2, const NodePtr &antiquant_scale1, const NodePtr &antiquant_scale2, const NodePtr &antiquant_offset1, const NodePtr &antiquant_offset2, const NodePtr &activation, const NodePtr &inner_precise);
-    static NodePtr Conv2DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr Equal(const NodePtr &input, const NodePtr &other);
+    static NodePtr MinDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr GroupNormGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &mean, const NodePtr &rstd, const NodePtr &gamma_opt, const NodePtr &num_groups, const NodePtr &dx_is_require, const NodePtr &dgamma_is_require, const NodePtr &dbeta_is_require);
+    static NodePtr Div(const NodePtr &input, const NodePtr &other);
+    static NodePtr ReduceAny(const NodePtr &x, const NodePtr &axis, const NodePtr &keep_dims);
+    static NodePtr Chunk(const NodePtr &input, const NodePtr &chunks, const NodePtr &dim);
+    static NodePtr InplaceRandom(const NodePtr &input, const NodePtr &from_, const NodePtr &to, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr LogicalXor(const NodePtr &input, const NodePtr &other);
+    static NodePtr NewEmpty(const NodePtr &input, const NodePtr &size, const NodePtr &dtype, const NodePtr &device);
+    static NodePtr LogSoftmaxGrad(const NodePtr &logits, const NodePtr &grad, const NodePtr &axis);
+    static NodePtr Frac(const NodePtr &input);
+    static NodePtr InplaceScatterValueReduce(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value, const NodePtr &reduce);
+    static NodePtr SeluGrad(const NodePtr &gradient, const NodePtr &result);
     static NodePtr Take(const NodePtr &input, const NodePtr &index);
-    static NodePtr AvgPool2DGrad(const NodePtr &grad, const NodePtr &image, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
-    static NodePtr Eye(const NodePtr &n, const NodePtr &m, const NodePtr &dtype);
-    static NodePtr AddRmsNorm(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &epsilon);
-    static NodePtr MaxUnpool2DExt(const NodePtr &input, const NodePtr &indices, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &output_size);
-    static NodePtr Embedding(const NodePtr &input, const NodePtr &weight, const NodePtr &padding_idx, const NodePtr &max_norm, const NodePtr &norm_type, const NodePtr &scale_grad_by_freq);
-    static NodePtr SoftMarginLoss(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
-    static NodePtr AdamW(const NodePtr &var, const NodePtr &m, const NodePtr &v, const NodePtr &max_v, const NodePtr &gradient, const NodePtr &step, const NodePtr &lr, const NodePtr &beta1, const NodePtr &beta2, const NodePtr &decay, const NodePtr &eps, const NodePtr &amsgrad, const NodePtr &maximize);
-    static NodePtr Sinh(const NodePtr &input);
-    static NodePtr RmsNormGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &rstd, const NodePtr &gamma);
-    static NodePtr Cummax(const NodePtr &input, const NodePtr &axis);
+    static NodePtr BincountExt(const NodePtr &input, const NodePtr &weights, const NodePtr &minlength);
+    static NodePtr AddLayerNormV2(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &beta, const NodePtr &epsilon, const NodePtr &additionalOut);
+    static NodePtr InplaceRemainderTensorTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr BitwiseXorTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr Sinc(const NodePtr &input);
+    static NodePtr BinaryCrossEntropyWithLogitsBackward(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &posWeight, const NodePtr &reduction);
+    static NodePtr HSigmoid(const NodePtr &input);
     static NodePtr Col2ImGrad(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &dilation, const NodePtr &padding, const NodePtr &stride);
-    static NodePtr AcosExt(const NodePtr &input);
-    static NodePtr MedianDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr InplaceAddsExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr ReplicationPad2DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
-    static NodePtr MinDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr GridSampler2D(const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners);
-    static NodePtr InplaceSubExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr Cast(const NodePtr &input, const NodePtr &dtype);
-    static NodePtr SilentCheckV2(const NodePtr &val, const NodePtr &input_grad, const NodePtr &sfda, const NodePtr &step, const NodePtr &c_min_steps, const NodePtr &c_thresh_l1, const NodePtr &c_coeff_l1, const NodePtr &c_thresh_l2, const NodePtr &c_coeff_l2, const NodePtr &npu_asd_detect);
-    static NodePtr InplaceDivMods(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
-    static NodePtr MaskedFill(const NodePtr &input_x, const NodePtr &mask, const NodePtr &value);
-    static NodePtr CumsumExt(const NodePtr &input, const NodePtr &dim, const NodePtr &dtype);
-    static NodePtr ReluGrad(const NodePtr &y_backprop, const NodePtr &x);
-    static NodePtr LogSumExp(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr BernoulliExt(const NodePtr &input, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr GeluGradExt(const NodePtr &grad, const NodePtr &input, const NodePtr &approximate);
-    static NodePtr IndexAddExt(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &source, const NodePtr &alpha);
-    static NodePtr InplaceMaskedFillTensor(const NodePtr &input, const NodePtr &mask, const NodePtr &value);
-    static NodePtr UpsampleTrilinear3D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr TanhGrad(const NodePtr &y, const NodePtr &dy);
-    static NodePtr InplaceLog(const NodePtr &input);
-    static NodePtr HSigmoidGrad(const NodePtr &grads, const NodePtr &input_x);
-    static NodePtr HSwish(const NodePtr &input);
-    static NodePtr SplitWithSizeView(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
-    static NodePtr Triu(const NodePtr &input, const NodePtr &diagonal);
-    static NodePtr LinSpaceExt(const NodePtr &start, const NodePtr &end, const NodePtr &steps, const NodePtr &dtype);
-    static NodePtr NonZeroExt(const NodePtr &input);
-    static NodePtr NewOnes(const NodePtr &input, const NodePtr &size, const NodePtr &dtype);
-    static NodePtr ZerosLikeExt(const NodePtr &input, const NodePtr &dtype);
-    static NodePtr Ones(const NodePtr &shape, const NodePtr &dtype);
-    static NodePtr IncreFlashAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &attn_mask, const NodePtr &actual_seq_lengths, const NodePtr &pse_shift, const NodePtr &dequant_scale1, const NodePtr &quant_scale1, const NodePtr &dequant_scale2, const NodePtr &quant_scale2, const NodePtr &quant_offset2, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &block_table, const NodePtr &kv_padding_size, const NodePtr &num_heads, const NodePtr &input_layout, const NodePtr &scale_value, const NodePtr &num_key_value_heads, const NodePtr &block_size, const NodePtr &inner_precise);
-    static NodePtr SmoothL1LossGrad(const NodePtr &prediction, const NodePtr &target, const NodePtr &dout, const NodePtr &beta, const NodePtr &reduction);
-    static NodePtr Narrow(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &length);
-    static NodePtr Transpose(const NodePtr &input, const NodePtr &input_perm);
-    static NodePtr Dot(const NodePtr &input, const NodePtr &other);
-    static NodePtr Tile(const NodePtr &input, const NodePtr &dims);
-    static NodePtr Pow(const NodePtr &input, const NodePtr &exponent);
-    static NodePtr MSELossGradExt(const NodePtr &dout, const NodePtr &x, const NodePtr &target, const NodePtr &reduction);
-    static NodePtr Floor(const NodePtr &input);
-    static NodePtr InplaceExp(const NodePtr &input);
-    static NodePtr TrilExt(const NodePtr &input, const NodePtr &diagonal);
-    static NodePtr DropoutGradExt(const NodePtr &input, const NodePtr &mask, const NodePtr &p);
-    static NodePtr InnerCommIsend(const NodePtr &input, const NodePtr &dst, const NodePtr &group, const NodePtr &tag);
-    static NodePtr Erfinv(const NodePtr &input);
-    static NodePtr DistCommReduceScatterTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
-    static NodePtr Tanh(const NodePtr &input);
-    static NodePtr LogicalOr(const NodePtr &x, const NodePtr &y);
-    static NodePtr Softmax(const NodePtr &input, const NodePtr &axis);
-    static NodePtr LinalgVectorNorm(const NodePtr &x, const NodePtr &ord, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr Atanh(const NodePtr &input);
-    static NodePtr Clone(const NodePtr &input);
-    static NodePtr DivMod(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
-    static NodePtr SumExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr AvgPool3DExt(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
-    static NodePtr ReplicationPad3D(const NodePtr &input, const NodePtr &padding);
-    static NodePtr InplaceAddExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr Polar(const NodePtr &abs, const NodePtr &angle);
-    static NodePtr Erfc(const NodePtr &input);
-    static NodePtr LogAddExp2(const NodePtr &input, const NodePtr &other);
-    static NodePtr SpeedFusionAttentionGrad(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &dy, const NodePtr &head_num, const NodePtr &input_layout, const NodePtr &pse, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &softmax_max, const NodePtr &softmax_sum, const NodePtr &softmax_in, const NodePtr &attention_in, const NodePtr &scale_value, const NodePtr &keep_prob, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &seed, const NodePtr &offset, const NodePtr &numels, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &sparse_mode, const NodePtr &gen_mask_parallel, const NodePtr &sync, const NodePtr &pse_type, const NodePtr &q_start_idx, const NodePtr &kv_start_idx);
-    static NodePtr EqualExt(const NodePtr &input, const NodePtr &other);
-    static NodePtr AdaptiveAvgPool3DExt(const NodePtr &input, const NodePtr &output_size);
-    static NodePtr DistCommAllToAllVSingle(const NodePtr &other, const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size, const NodePtr &split_sizes_empty);
-    static NodePtr XLogYScalarSelf(const NodePtr &input, const NodePtr &other);
-    static NodePtr DistCommAllGather(const NodePtr &tensor_list, const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr LayerNormExt(const NodePtr &input, const NodePtr &normalized_shape, const NodePtr &weight, const NodePtr &bias, const NodePtr &eps);
+    static NodePtr AvgPool1D(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad);
+    static NodePtr Conv2DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr MeanExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr Cross(const NodePtr &input, const NodePtr &other, const NodePtr &dim);
+    static NodePtr NewZeros(const NodePtr &input, const NodePtr &size, const NodePtr &dtype);
+    static NodePtr Atan2Ext(const NodePtr &input, const NodePtr &other);
+    static NodePtr EmbeddingDenseBackward(const NodePtr &grad, const NodePtr &indices, const NodePtr &num_weights, const NodePtr &padding_idx, const NodePtr &scale_grad_by_freq);
+    static NodePtr FmodTensor(const NodePtr &input, const NodePtr &other);
+    static NodePtr BinaryCrossEntropy(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction);
     static NodePtr DistCommIsend(const NodePtr &input, const NodePtr &dst, const NodePtr &group, const NodePtr &tag);
-    static NodePtr KLDiv(const NodePtr &input, const NodePtr &target, const NodePtr &reduction, const NodePtr &log_target);
-    static NodePtr SubExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
-    static NodePtr MultiScaleDeformableAttn(const NodePtr &value, const NodePtr &shape, const NodePtr &offset, const NodePtr &locations, const NodePtr &weight);
-    static NodePtr Im2ColExt(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &dilation, const NodePtr &padding, const NodePtr &stride);
-    static NodePtr InplaceFillDiagonal(const NodePtr &input, const NodePtr &fill_value, const NodePtr &wrap);
-    static NodePtr DiagExt(const NodePtr &input, const NodePtr &diagonal);
+    static NodePtr GridSampler3D(const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners);
+    static NodePtr StackExt(const NodePtr &tensors, const NodePtr &dim);
     static NodePtr ReduceMin(const NodePtr &x, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr PReLU(const NodePtr &input, const NodePtr &weight);
-    static NodePtr NanToNum(const NodePtr &input, const NodePtr &nan, const NodePtr &posinf, const NodePtr &neginf);
-    static NodePtr CumminExt(const NodePtr &input, const NodePtr &dim);
-    static NodePtr SiLUGrad(const NodePtr &dout, const NodePtr &x);
-    static NodePtr InnerIndex(const NodePtr &input, const NodePtr &indices);
-    static NodePtr KLDivGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &reduction, const NodePtr &log_target);
-    static NodePtr NormalFloatTensor(const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr HistcExt(const NodePtr &input, const NodePtr &bins, const NodePtr &min, const NodePtr &max);
-    static NodePtr LessEqual(const NodePtr &input, const NodePtr &other);
-    static NodePtr EluExt(const NodePtr &input, const NodePtr &alpha);
-    static NodePtr Outer(const NodePtr &input, const NodePtr &vec2);
-    static NodePtr Squeeze(const NodePtr &input, const NodePtr &axis);
-    static NodePtr ArgSort(const NodePtr &input, const NodePtr &dim, const NodePtr &descending, const NodePtr &stable);
-    static NodePtr AddLayerNormGrad(const NodePtr &dy, const NodePtr &x1, const NodePtr &x2, const NodePtr &rstd, const NodePtr &mean, const NodePtr &gamma, const NodePtr &dsumOptional);
-    static NodePtr ProdExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr BatchMatMul(const NodePtr &x, const NodePtr &y, const NodePtr &transpose_a, const NodePtr &transpose_b);
-    static NodePtr Swiglu(const NodePtr &input, const NodePtr &dim);
-    static NodePtr RemainderTensorTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr InnerNonZero(const NodePtr &input);
-    static NodePtr Gcd(const NodePtr &input, const NodePtr &other);
-    static NodePtr UpsampleBilinear2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr NLLLoss2d(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index);
-    static NodePtr Trunc(const NodePtr &input);
-    static NodePtr GeluExt(const NodePtr &input, const NodePtr &approximate);
-    static NodePtr Div(const NodePtr &input, const NodePtr &other);
-    static NodePtr AllFinite(const NodePtr &tensors);
-    static NodePtr DistCommGatherIntoTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &dst, const NodePtr &rank_id, const NodePtr &group);
+    static NodePtr InplaceClampTensor(const NodePtr &input, const NodePtr &min, const NodePtr &max);
+    static NodePtr RepeatInterleaveTensor(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim, const NodePtr &output_size);
+    static NodePtr GluGrad(const NodePtr &grads, const NodePtr &x, const NodePtr &axis);
     static NodePtr DistCommReduce(const NodePtr &input, const NodePtr &op_type, const NodePtr &dst, const NodePtr &group);
-    static NodePtr Expm1(const NodePtr &input);
-    static NodePtr MaskedSelectGrad(const NodePtr &input, const NodePtr &mask, const NodePtr &grad);
+    static NodePtr ConvolutionGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &output_mask);
+    static NodePtr InplaceSubExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr MishGradExt(const NodePtr &dout, const NodePtr &x);
+    static NodePtr TanhGrad(const NodePtr &y, const NodePtr &dy);
+    static NodePtr BatchMatMulExt(const NodePtr &input, const NodePtr &mat2);
+    static NodePtr IndexSelect(const NodePtr &input, const NodePtr &dim, const NodePtr &index);
+    static NodePtr UnstackExtView(const NodePtr &input, const NodePtr &dim);
+    static NodePtr DistCommAllGatherIntoTensorUneven(const NodePtr &other, const NodePtr &input, const NodePtr &output_split_sizes, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr InplaceFillTensor(const NodePtr &input, const NodePtr &value);
+    static NodePtr LogSumExp(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr CrossEntropyLoss(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &reduction, const NodePtr &ignore_index, const NodePtr &label_smoothing, const NodePtr &lse_square_scale_for_zloss, const NodePtr &return_zloss);
+    static NodePtr NonZero(const NodePtr &input);
+    static NodePtr PowTensorScalar(const NodePtr &input, const NodePtr &exponent);
+    static NodePtr InplaceMatmulAdd(const NodePtr &x, const NodePtr &weight, const NodePtr &C);
+    static NodePtr TypeAs(const NodePtr &input, const NodePtr &other);
+    static NodePtr Polar(const NodePtr &abs, const NodePtr &angle);
+    static NodePtr DistCommIrecv(const NodePtr &input, const NodePtr &tag, const NodePtr &src, const NodePtr &group);
+    static NodePtr FillTensor(const NodePtr &size, const NodePtr &fill_value, const NodePtr &dtype);
+    static NodePtr Tanh(const NodePtr &input);
+    static NodePtr Empty(const NodePtr &size, const NodePtr &dtype, const NodePtr &device);
+    static NodePtr Xlogy(const NodePtr &input, const NodePtr &other);
+    static NodePtr FloorDivScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr ArgSort(const NodePtr &input, const NodePtr &dim, const NodePtr &descending, const NodePtr &stable);
+    static NodePtr L1LossBackwardExt(const NodePtr &grad_output, const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
+    static NodePtr InplaceScatterValue(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value);
+    static NodePtr InplaceHardtanh(const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
+    static NodePtr DivMods(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
+    static NodePtr ViewAs(const NodePtr &input, const NodePtr &other);
+    static NodePtr Sinh(const NodePtr &input);
+    static NodePtr SoftShrinkGrad(const NodePtr &input_grad, const NodePtr &input_x, const NodePtr &lambd);
+    static NodePtr InplaceBernoulliScalar(const NodePtr &input, const NodePtr &p, const NodePtr &seed, const NodePtr &offset);
+    static NodePtr GeLU(const NodePtr &input);
     static NodePtr DistCommScatter(const NodePtr &other, const NodePtr &scatter_list, const NodePtr &rank_size, const NodePtr &src, const NodePtr &rank_id, const NodePtr &group);
-    static NodePtr SmoothL1Loss(const NodePtr &prediction, const NodePtr &target, const NodePtr &beta, const NodePtr &reduction);
-    static NodePtr SeLUExt(const NodePtr &input);
-    static NodePtr Mv(const NodePtr &input, const NodePtr &vec);
-    static NodePtr MedianExt(const NodePtr &input);
+    static NodePtr Log2(const NodePtr &input);
+    static NodePtr SoftMarginLossGrad(const NodePtr &predict, const NodePtr &label, const NodePtr &dout, const NodePtr &reduction);
+    static NodePtr RmsNormGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &rstd, const NodePtr &gamma);
+    static NodePtr Nansum(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr SelectExtView(const NodePtr &input, const NodePtr &dim, const NodePtr &index);
+    static NodePtr StdMean(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
+    static NodePtr UniqueConsecutive(const NodePtr &input, const NodePtr &return_inverse, const NodePtr &return_counts, const NodePtr &dim);
+    static NodePtr TriangularSolve(const NodePtr &b, const NodePtr &A, const NodePtr &upper, const NodePtr &transpose, const NodePtr &unitriangular);
+    static NodePtr PagedAttention(const NodePtr &query, const NodePtr &key_cache, const NodePtr &value_cache, const NodePtr &block_tables, const NodePtr &context_lens, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &attn_mask, const NodePtr &q_seq_lens, const NodePtr &alibi_mask, const NodePtr &head_num, const NodePtr &scale_value, const NodePtr &kv_head_num, const NodePtr &kv_cache_quant_mode, const NodePtr &mask_mode, const NodePtr &mla_v_dim);
+    static NodePtr RingAttentionUpdate(const NodePtr &prev_attn_out, const NodePtr &prev_softmax_max, const NodePtr &prev_softmax_sum, const NodePtr &cur_attn_out, const NodePtr &cur_softmax_max, const NodePtr &cur_softmax_sum, const NodePtr &actual_seq_qlen, const NodePtr &layout);
+    static NodePtr AddExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr AdaptiveAvgPool3DExt(const NodePtr &input, const NodePtr &output_size);
+    static NodePtr SubExt(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr InplaceExp(const NodePtr &input);
+    static NodePtr Greater(const NodePtr &input, const NodePtr &other);
     static NodePtr IndexFillTensor(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &value);
-    static NodePtr AddScalar(const NodePtr &input, const NodePtr &other, const NodePtr &alpha);
+    static NodePtr LayerNormGradExt(const NodePtr &dy, const NodePtr &x, const NodePtr &normalized_shape, const NodePtr &mean, const NodePtr &variance, const NodePtr &gamma, const NodePtr &beta, const NodePtr &output_mask);
+    static NodePtr GatherDGradV2(const NodePtr &x, const NodePtr &dim, const NodePtr &index, const NodePtr &dout);
+    static NodePtr InnerCommAllReduce(const NodePtr &input, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr AddLayerNormGrad(const NodePtr &dy, const NodePtr &x1, const NodePtr &x2, const NodePtr &rstd, const NodePtr &mean, const NodePtr &gamma, const NodePtr &dsumOptional);
+    static NodePtr Convolution(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups);
+    static NodePtr MultiScaleDeformableAttn(const NodePtr &value, const NodePtr &shape, const NodePtr &offset, const NodePtr &locations, const NodePtr &weight);
+    static NodePtr InplaceFillScalar(const NodePtr &input, const NodePtr &value);
+    static NodePtr InplaceAddmm(const NodePtr &input, const NodePtr &mat1, const NodePtr &mat2, const NodePtr &beta, const NodePtr &alpha);
+    static NodePtr ReplicationPad3DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
+    static NodePtr GridSampler2D(const NodePtr &input_x, const NodePtr &grid, const NodePtr &interpolation_mode, const NodePtr &padding_mode, const NodePtr &align_corners);
+    static NodePtr Reshape(const NodePtr &input, const NodePtr &shape);
+    static NodePtr DiagonalView(const NodePtr &input, const NodePtr &offset, const NodePtr &dim1, const NodePtr &dim2);
+    static NodePtr Conv3DExt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
+    static NodePtr Addmv(const NodePtr &input, const NodePtr &mat, const NodePtr &vec, const NodePtr &beta, const NodePtr &alpha);
+    static NodePtr UpsampleLinear1D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr BCEWithLogitsLoss(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &posWeight, const NodePtr &reduction);
+    static NodePtr UpsampleNearest1D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr InplaceElu(const NodePtr &input, const NodePtr &alpha);
+    static NodePtr LeakyReLUExt(const NodePtr &input, const NodePtr &negative_slope);
+    static NodePtr Scatter(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
     static NodePtr RotaryPositionEmbeddingGrad(const NodePtr &dy, const NodePtr &cos, const NodePtr &sin, const NodePtr &dx, const NodePtr &mode);
-    static NodePtr UpsampleNearest3D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr Abs(const NodePtr &input);
-    static NodePtr ConvolutionStrGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &output_mask);
-    static NodePtr Erf(const NodePtr &input);
+    static NodePtr Index(const NodePtr &input, const NodePtr &indices);
+    static NodePtr MaxUnpool2DExt(const NodePtr &input, const NodePtr &indices, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &output_size);
+    static NodePtr MedianDim(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr RandInt(const NodePtr &low, const NodePtr &high, const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr Round(const NodePtr &input, const NodePtr &decimals);
+    static NodePtr ProdExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr UpsampleNearest1DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr BinaryCrossEntropyGrad(const NodePtr &input, const NodePtr &target, const NodePtr &grad_output, const NodePtr &weight, const NodePtr &reduction);
+    static NodePtr InnerCommIrecv(const NodePtr &tag, const NodePtr &src, const NodePtr &shape, const NodePtr &group, const NodePtr &dtype);
+    static NodePtr AdaptiveAvgPool2DGradExt(const NodePtr &grad_output, const NodePtr &x);
+    static NodePtr IndexAddExt(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &source, const NodePtr &alpha);
+    static NodePtr Divs(const NodePtr &input, const NodePtr &other);
+    static NodePtr BatchNormElemt(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &mean, const NodePtr &invstd, const NodePtr &eps);
+    static NodePtr Std(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
+    static NodePtr UpsampleBicubic2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr Threshold(const NodePtr &input, const NodePtr &threshold, const NodePtr &value);
+    static NodePtr AddRmsNorm(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &epsilon);
     static NodePtr Sigmoid(const NodePtr &input);
-    static NodePtr UpsampleNearest2D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales);
-    static NodePtr MultinomialExt(const NodePtr &input, const NodePtr &num_samples, const NodePtr &replacement, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr InplaceScatterAdd(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
+    static NodePtr ReverseV2(const NodePtr &input, const NodePtr &axis);
+    static NodePtr InnerCommIsend(const NodePtr &input, const NodePtr &dst, const NodePtr &group, const NodePtr &tag);
+    static NodePtr SliceExt(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &end, const NodePtr &step);
+    static NodePtr SilentCheckV3(const NodePtr &val, const NodePtr &max, const NodePtr &avg, const NodePtr &input_grad, const NodePtr &step, const NodePtr &c_thresh_l1, const NodePtr &c_thresh_l2, const NodePtr &beta1, const NodePtr &npu_asd_detect);
+    static NodePtr TrilExt(const NodePtr &input, const NodePtr &diagonal);
+    static NodePtr DistCommReduceScatterTensorUneven(const NodePtr &other, const NodePtr &input, const NodePtr &input_split_size, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr NarrowView(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &length);
+    static NodePtr AvgPool2DGrad(const NodePtr &grad, const NodePtr &image, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
     static NodePtr InplaceGroupedMatmulAdd(const NodePtr &x, const NodePtr &weight, const NodePtr &group_list, const NodePtr &out);
-    static NodePtr GeLU(const NodePtr &input);
-    static NodePtr DropoutGenMaskExt(const NodePtr &shape, const NodePtr &p, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr Sign(const NodePtr &input);
-    static NodePtr ArgMaxExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr ThresholdGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &threshold);
-    static NodePtr AvgPool2D(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &ceil_mode, const NodePtr &count_include_pad, const NodePtr &divisor_override);
-    static NodePtr ReduceAny(const NodePtr &x, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr LeakyReLUGradExt(const NodePtr &dy, const NodePtr &input, const NodePtr &negative_slope, const NodePtr &is_result);
+    static NodePtr Hardtanh(const NodePtr &input, const NodePtr &min_val, const NodePtr &max_val);
+    static NodePtr LessEqual(const NodePtr &input, const NodePtr &other);
+    static NodePtr MoeDistributeCombine(const NodePtr &expand_x, const NodePtr &expert_ids, const NodePtr &expand_idx, const NodePtr &ep_send_counts, const NodePtr &expert_scales, const NodePtr &ep_world_size, const NodePtr &ep_rank_id, const NodePtr &moe_expert_num, const NodePtr &tp_send_counts, const NodePtr &x_active_mask, const NodePtr &activate_scale, const NodePtr &weight_scale, const NodePtr &group_list, const NodePtr &expand_scales, const NodePtr &group_ep, const NodePtr &group_tp, const NodePtr &tp_world_size, const NodePtr &tp_rank_id, const NodePtr &expert_shard_type, const NodePtr &shared_expert_num, const NodePtr &shared_export_rank_num, const NodePtr &global_bs, const NodePtr &out_dtype, const NodePtr &common_quant_mode, const NodePtr &group_list_type);
+    static NodePtr Tile(const NodePtr &input, const NodePtr &dims);
+    static NodePtr TopkExt(const NodePtr &input, const NodePtr &k, const NodePtr &dim, const NodePtr &largest, const NodePtr &sorted);
+    static NodePtr InnerCommReduceScatter(const NodePtr &input, const NodePtr &rank_size, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr AtanExt(const NodePtr &input);
     static NodePtr AdaptiveAvgPool2DExt(const NodePtr &input, const NodePtr &output_size);
-    static NodePtr GreaterEqualScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr Cos(const NodePtr &input);
+    static NodePtr ReplicationPad2DGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &padding);
+    static NodePtr DistCommAllGatherIntoTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &group);
+    static NodePtr DropoutGenMaskExt(const NodePtr &shape, const NodePtr &p, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr DivMod(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
+    static NodePtr KLDiv(const NodePtr &input, const NodePtr &target, const NodePtr &reduction, const NodePtr &log_target);
+    static NodePtr Cosh(const NodePtr &input);
+    static NodePtr SilentCheckV2(const NodePtr &val, const NodePtr &input_grad, const NodePtr &sfda, const NodePtr &step, const NodePtr &c_min_steps, const NodePtr &c_thresh_l1, const NodePtr &c_coeff_l1, const NodePtr &c_thresh_l2, const NodePtr &c_coeff_l2, const NodePtr &npu_asd_detect);
+    static NodePtr BatchNormGatherStatsWithCounts(const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &running_mean, const NodePtr &running_var, const NodePtr &momentum, const NodePtr &eps, const NodePtr &counts);
+    static NodePtr XLogYScalarSelf(const NodePtr &input, const NodePtr &other);
+    static NodePtr LogicalAnd(const NodePtr &x, const NodePtr &y);
+    static NodePtr InnerCommAllToAllV(const NodePtr &input, const NodePtr &group, const NodePtr &send_numel_list, const NodePtr &recv_numel_list, const NodePtr &rank_size, const NodePtr &split_sizes_empty);
+    static NodePtr FillScalar(const NodePtr &size, const NodePtr &fill_value, const NodePtr &dtype);
+    static NodePtr Gcd(const NodePtr &input, const NodePtr &other);
+    static NodePtr Min(const NodePtr &input);
     static NodePtr UpsampleBicubic2D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr EmbeddingDenseBackward(const NodePtr &grad, const NodePtr &indices, const NodePtr &num_weights, const NodePtr &padding_idx, const NodePtr &scale_grad_by_freq);
+    static NodePtr ConvolutionStr(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups);
+    static NodePtr UpsampleLinear1DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr GeluExt(const NodePtr &input, const NodePtr &approximate);
     static NodePtr UpsampleTrilinear3DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr RandnLike(const NodePtr &input, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
-    static NodePtr MaxPoolGradWithMask(const NodePtr &x, const NodePtr &grad, const NodePtr &mask, const NodePtr &kernel_size, const NodePtr &strides, const NodePtr &pads, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &argmax_type);
-    static NodePtr ArgMaxWithValue(const NodePtr &input, const NodePtr &axis, const NodePtr &keep_dims);
-    static NodePtr GatherD(const NodePtr &x, const NodePtr &dim, const NodePtr &index);
-    static NodePtr DistCommScatterTensor(const NodePtr &other, const NodePtr &input, const NodePtr &rank_size, const NodePtr &src, const NodePtr &rank_id, const NodePtr &group);
-    static NodePtr ConvolutionGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &transposed, const NodePtr &output_padding, const NodePtr &groups, const NodePtr &output_mask);
-    static NodePtr AcoshExt(const NodePtr &input);
-    static NodePtr InplaceMul(const NodePtr &input, const NodePtr &other);
-    static NodePtr InplaceClampScalar(const NodePtr &input, const NodePtr &min, const NodePtr &max);
-    static NodePtr IndexSelect(const NodePtr &input, const NodePtr &dim, const NodePtr &index);
-    static NodePtr InplaceUniform(const NodePtr &input, const NodePtr &from_, const NodePtr &to, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr Rsqrt(const NodePtr &input);
-    static NodePtr Scatter(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
-    static NodePtr XLogYScalarOther(const NodePtr &input, const NodePtr &other);
-    static NodePtr Sub(const NodePtr &input, const NodePtr &other);
-    static NodePtr Xlogy(const NodePtr &input, const NodePtr &other);
-    static NodePtr NormalFloatFloat(const NodePtr &mean, const NodePtr &std, const NodePtr &size, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr TransposeView(const NodePtr &input, const NodePtr &input_perm);
-    static NodePtr FillScalar(const NodePtr &size, const NodePtr &fill_value, const NodePtr &dtype);
-    static NodePtr UniqueConsecutive(const NodePtr &input, const NodePtr &return_inverse, const NodePtr &return_counts, const NodePtr &dim);
-    static NodePtr RotaryPositionEmbedding(const NodePtr &x, const NodePtr &cos, const NodePtr &sin, const NodePtr &mode);
-    static NodePtr SearchSorted(const NodePtr &sorted_sequence, const NodePtr &values, const NodePtr &sorter, const NodePtr &dtype, const NodePtr &right);
-    static NodePtr Norm(const NodePtr &input, const NodePtr &p, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
-    static NodePtr NormalTensorTensor(const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr Atan2Ext(const NodePtr &input, const NodePtr &other);
-    static NodePtr NLLLossGrad(const NodePtr &logits, const NodePtr &loss_grad, const NodePtr &labels, const NodePtr &weight, const NodePtr &total_weight, const NodePtr &reduction, const NodePtr &ignore_index);
-    static NodePtr SplitTensorView(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
-    static NodePtr SilentCheckV3(const NodePtr &val, const NodePtr &max, const NodePtr &avg, const NodePtr &input_grad, const NodePtr &step, const NodePtr &c_thresh_l1, const NodePtr &c_thresh_l2, const NodePtr &beta1, const NodePtr &npu_asd_detect);
-    static NodePtr Conv3DPadding(const NodePtr &input, const NodePtr &weight, const NodePtr &bias, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &groups);
-    static NodePtr DivMods(const NodePtr &input, const NodePtr &other, const NodePtr &rounding_mode);
-    static NodePtr LogicalXor(const NodePtr &input, const NodePtr &other);
-    static NodePtr DistCommBroadcast(const NodePtr &input, const NodePtr &src, const NodePtr &rank_id, const NodePtr &group);
+    static NodePtr UniqueDim(const NodePtr &input, const NodePtr &sorted, const NodePtr &return_inverse, const NodePtr &dim);
+    static NodePtr VarMean(const NodePtr &input, const NodePtr &dim, const NodePtr &correction, const NodePtr &keepdim);
+    static NodePtr GreaterEqualScalar(const NodePtr &input, const NodePtr &other);
+    static NodePtr HSigmoidGrad(const NodePtr &grads, const NodePtr &input_x);
+    static NodePtr CumminExt(const NodePtr &input, const NodePtr &dim);
+    static NodePtr SoftMarginLoss(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
+    static NodePtr Square(const NodePtr &input);
+    static NodePtr UpsampleBilinear2D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr Tan(const NodePtr &input);
+    static NodePtr SplitWithSize(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
+    static NodePtr DistCommBarrier(const NodePtr &group);
+    static NodePtr Mul(const NodePtr &input, const NodePtr &other);
+    static NodePtr ThresholdGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &threshold);
+    static NodePtr SelectV2(const NodePtr &condition, const NodePtr &input, const NodePtr &other);
+    static NodePtr Erfinv(const NodePtr &input);
+    static NodePtr Sin(const NodePtr &input);
+    static NodePtr DropoutGradExt(const NodePtr &input, const NodePtr &mask, const NodePtr &p);
+    static NodePtr Mm(const NodePtr &input, const NodePtr &mat2);
+    static NodePtr LogAddExp2(const NodePtr &input, const NodePtr &other);
+    static NodePtr PReLUGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &weight);
+    static NodePtr PromptFlashAttention(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &attn_mask, const NodePtr &actual_seq_lengths, const NodePtr &actual_seq_lengths_kv, const NodePtr &pse_shift, const NodePtr &deq_scale1, const NodePtr &quant_scale1, const NodePtr &deq_scale2, const NodePtr &quant_scale2, const NodePtr &quant_offset2, const NodePtr &num_heads, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &input_layout, const NodePtr &num_key_value_heads, const NodePtr &sparse_mode, const NodePtr &inner_precise);
+    static NodePtr Im2ColExt(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &dilation, const NodePtr &padding, const NodePtr &stride);
+    static NodePtr Split(const NodePtr &input_x, const NodePtr &axis, const NodePtr &output_num);
+    static NodePtr Narrow(const NodePtr &input, const NodePtr &dim, const NodePtr &start, const NodePtr &length);
+    static NodePtr Trunc(const NodePtr &input);
     static NodePtr MoeTokenPermute(const NodePtr &tokens, const NodePtr &indices, const NodePtr &num_out_tokens, const NodePtr &padded_mode);
     static NodePtr NormalTensorFloat(const NodePtr &mean, const NodePtr &std, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr MatmulReduceScatter(const NodePtr &input, const NodePtr &x2, const NodePtr &group, const NodePtr &world_size, const NodePtr &reduce_op, const NodePtr &bias, const NodePtr &comm_turn, const NodePtr &trans_input, const NodePtr &trans_x2);
-    static NodePtr HSigmoid(const NodePtr &input);
-    static NodePtr Kthvalue(const NodePtr &input, const NodePtr &k, const NodePtr &dim, const NodePtr &keepdim);
-    static NodePtr ExpandDimsView(const NodePtr &input, const NodePtr &dim);
-    static NodePtr BatchNormStats(const NodePtr &input, const NodePtr &eps);
-    static NodePtr UpsampleBicubic2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr UpsampleLinear1DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr BitwiseOrTensor(const NodePtr &input, const NodePtr &other);
-    static NodePtr BCEWithLogitsLoss(const NodePtr &input, const NodePtr &target, const NodePtr &weight, const NodePtr &posWeight, const NodePtr &reduction);
-    static NodePtr LogSigmoid(const NodePtr &input);
-    static NodePtr GeLUGrad(const NodePtr &dy, const NodePtr &x, const NodePtr &y);
-    static NodePtr MSELossExt(const NodePtr &input, const NodePtr &target, const NodePtr &reduction);
-    static NodePtr MatMulExt(const NodePtr &input, const NodePtr &other);
-    static NodePtr MishGradExt(const NodePtr &dout, const NodePtr &x);
+    static NodePtr Eye(const NodePtr &n, const NodePtr &m, const NodePtr &dtype);
     static NodePtr BitwiseNot(const NodePtr &input);
-    static NodePtr UpsampleBilinear2D(const NodePtr &x, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
-    static NodePtr QuantV2(const NodePtr &x, const NodePtr &scale, const NodePtr &offset, const NodePtr &sqrt_mode, const NodePtr &rounding_mode, const NodePtr &dst_type);
+    static NodePtr MishExt(const NodePtr &input);
+    static NodePtr FFNExt(const NodePtr &x, const NodePtr &weight1, const NodePtr &weight2, const NodePtr &expertTokens, const NodePtr &bias1, const NodePtr &bias2, const NodePtr &scale, const NodePtr &offset, const NodePtr &deqScale1, const NodePtr &deqScale2, const NodePtr &antiquant_scale1, const NodePtr &antiquant_scale2, const NodePtr &antiquant_offset1, const NodePtr &antiquant_offset2, const NodePtr &activation, const NodePtr &inner_precise);
+    static NodePtr ChunkView(const NodePtr &input, const NodePtr &chunks, const NodePtr &dim);
+    static NodePtr Log1p(const NodePtr &input);
+    static NodePtr SwigluGrad(const NodePtr &grad_output, const NodePtr &input, const NodePtr &dim);
+    static NodePtr BatchNormElemtGrad(const NodePtr &dout, const NodePtr &input, const NodePtr &mean, const NodePtr &invstd, const NodePtr &weight, const NodePtr &sumd_dy, const NodePtr &sum_dy_xmu, const NodePtr &count);
+    static NodePtr Roll(const NodePtr &input, const NodePtr &shifts, const NodePtr &dims);
+    static NodePtr Addbmm(const NodePtr &input, const NodePtr &batch1, const NodePtr &batch2, const NodePtr &beta, const NodePtr &alpha);
+    static NodePtr Norm(const NodePtr &input, const NodePtr &p, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr InplaceMaskedFillScalar(const NodePtr &input, const NodePtr &mask, const NodePtr &value);
+    static NodePtr HistcExt(const NodePtr &input, const NodePtr &bins, const NodePtr &min, const NodePtr &max);
+    static NodePtr SplitTensor(const NodePtr &input, const NodePtr &split_size, const NodePtr &dim);
+    static NodePtr Erf(const NodePtr &input);
+    static NodePtr ScatterAddExt(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src);
+    static NodePtr InplaceScatterSrcReduce(const NodePtr &input, const NodePtr &dim, const NodePtr &index, const NodePtr &src, const NodePtr &reduce);
+    static NodePtr DistCommAllReduce(const NodePtr &input, const NodePtr &op_type, const NodePtr &group);
+    static NodePtr RandExt(const NodePtr &shape, const NodePtr &seed, const NodePtr &offset, const NodePtr &dtype);
+    static NodePtr Concat(const NodePtr &tensors, const NodePtr &axis);
+    static NodePtr MatMul(const NodePtr &input, const NodePtr &mat2, const NodePtr &transpose_a, const NodePtr &transpose_b);
+    static NodePtr ArgMaxExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr LogicalNot(const NodePtr &input);
+    static NodePtr GeluGradExt(const NodePtr &grad, const NodePtr &input, const NodePtr &approximate);
+    static NodePtr TraceExt(const NodePtr &input);
+    static NodePtr MatrixInverseExt(const NodePtr &input);
+    static NodePtr EmptyLike(const NodePtr &input, const NodePtr &dtype, const NodePtr &device);
+    static NodePtr Col2ImExt(const NodePtr &input, const NodePtr &output_size, const NodePtr &kernel_size, const NodePtr &dilation, const NodePtr &padding, const NodePtr &stride);
+    static NodePtr Dot(const NodePtr &input, const NodePtr &other);
+    static NodePtr SigmoidGrad(const NodePtr &y, const NodePtr &dy);
+    static NodePtr UpsampleBilinear2DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales, const NodePtr &align_corners);
+    static NodePtr Ones(const NodePtr &shape, const NodePtr &dtype);
+    static NodePtr NanToNum(const NodePtr &input, const NodePtr &nan, const NodePtr &posinf, const NodePtr &neginf);
+    static NodePtr DistCommBroadcast(const NodePtr &input, const NodePtr &src, const NodePtr &rank_id, const NodePtr &group);
+    static NodePtr SpeedFusionAttentionGrad(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &dy, const NodePtr &head_num, const NodePtr &input_layout, const NodePtr &pse, const NodePtr &padding_mask, const NodePtr &atten_mask, const NodePtr &softmax_max, const NodePtr &softmax_sum, const NodePtr &softmax_in, const NodePtr &attention_in, const NodePtr &scale_value, const NodePtr &keep_prob, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &inner_precise, const NodePtr &seed, const NodePtr &offset, const NodePtr &numels, const NodePtr &prefix, const NodePtr &actual_seq_qlen, const NodePtr &actual_seq_kvlen, const NodePtr &sparse_mode, const NodePtr &gen_mask_parallel, const NodePtr &sync, const NodePtr &pse_type, const NodePtr &q_start_idx, const NodePtr &kv_start_idx);
+    static NodePtr RepeatInterleaveGrad(const NodePtr &input, const NodePtr &repeats, const NodePtr &dim);
+    static NodePtr LogSigmoid(const NodePtr &input);
+    static NodePtr Reciprocal(const NodePtr &input);
+    static NodePtr AllFinite(const NodePtr &tensors);
+    static NodePtr UpsampleNearest3DGrad(const NodePtr &dy, const NodePtr &input_size, const NodePtr &output_size, const NodePtr &scales);
+    static NodePtr LinalgVectorNorm(const NodePtr &x, const NodePtr &ord, const NodePtr &dim, const NodePtr &keepdim, const NodePtr &dtype);
+    static NodePtr MoeFinalizeRouting(const NodePtr &expanded_x, const NodePtr &x1, const NodePtr &x2, const NodePtr &bias, const NodePtr &scales, const NodePtr &expanded_row_idx, const NodePtr &expanded_expert_idx);
     static NodePtr MoeGatingTopKSoftmax(const NodePtr &x, const NodePtr &finished, const NodePtr &k);
-    static NodePtr QuantBatchMatmul(const NodePtr &x1, const NodePtr &x2, const NodePtr &scale, const NodePtr &offset, const NodePtr &bias, const NodePtr &pertokenScaleOptional, const NodePtr &transpose_x1, const NodePtr &transpose_x2, const NodePtr &dtype);
+    static NodePtr QuantV2(const NodePtr &x, const NodePtr &scale, const NodePtr &offset, const NodePtr &sqrt_mode, const NodePtr &rounding_mode, const NodePtr &dst_type);
+    static NodePtr AddRmsNormQuantV2(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &scale, const NodePtr &offset, const NodePtr &epsilon);
+    static NodePtr MoeInitRoutingQuantV2(const NodePtr &x, const NodePtr &expert_idx, const NodePtr &active_num, const NodePtr &expert_capacity, const NodePtr &expert_num, const NodePtr &drop_pad_mode, const NodePtr &expert_tokens_count_or_cumsum_flag, const NodePtr &expert_tokens_before_capacity_flag, const NodePtr &quant_mode, const NodePtr &scale, const NodePtr &offset);
+    static NodePtr MoeInitRoutingV2(const NodePtr &x, const NodePtr &expert_idx, const NodePtr &active_num, const NodePtr &expert_capacity, const NodePtr &expert_num, const NodePtr &drop_pad_mode, const NodePtr &expert_tokens_count_or_cumsum_flag, const NodePtr &expert_tokens_before_capacity_flag);
+    static NodePtr MatmulAllReduceAddRmsNorm(const NodePtr &x1, const NodePtr &x2, const NodePtr &bias, const NodePtr &residual, const NodePtr &gamma, const NodePtr &epsilon, const NodePtr &group, const NodePtr &reduce_op, const NodePtr &comm_turn, const NodePtr &stream_mode);
+    static NodePtr GroupedMatmul(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &scale, const NodePtr &offset, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &group_list, const NodePtr &split_item, const NodePtr &group_type, const NodePtr &transpose_a, const NodePtr &transpose_b);
     static NodePtr KVCacheScatterUpdate(const NodePtr &var, const NodePtr &indices, const NodePtr &updates, const NodePtr &axis, const NodePtr &reduce);
     static NodePtr FusedInferAttentionScore(const NodePtr &query, const NodePtr &key, const NodePtr &value, const NodePtr &pse_shift, const NodePtr &attn_mask, const NodePtr &actual_seq_lengths, const NodePtr &actual_seq_lengths_kv, const NodePtr &dequant_scale1, const NodePtr &quant_scale1, const NodePtr &dequant_scale2, const NodePtr &quant_scale2, const NodePtr &quant_offset2, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &block_table, const NodePtr &query_padding_size, const NodePtr &kv_padding_size, const NodePtr &key_antiquant_scale, const NodePtr &key_antiquant_offset, const NodePtr &value_antiquant_scale, const NodePtr &value_antiquant_offset, const NodePtr &key_shared_prefix, const NodePtr &value_shared_prefix, const NodePtr &actual_shared_prefix_len, const NodePtr &num_heads, const NodePtr &scale_value, const NodePtr &pre_tokens, const NodePtr &next_tokens, const NodePtr &input_layout, const NodePtr &num_key_value_heads, const NodePtr &sparse_mode, const NodePtr &inner_precise, const NodePtr &block_size, const NodePtr &antiquant_mode, const NodePtr &softmax_lse_flag, const NodePtr &key_antiquant_mode, const NodePtr &value_antiquant_mode);
-    static NodePtr MatmulAllReduceAddRmsNorm(const NodePtr &x1, const NodePtr &x2, const NodePtr &bias, const NodePtr &residual, const NodePtr &gamma, const NodePtr &epsilon, const NodePtr &group, const NodePtr &reduce_op, const NodePtr &comm_turn, const NodePtr &stream_mode);
-    static NodePtr MoeFinalizeRouting(const NodePtr &expanded_x, const NodePtr &x1, const NodePtr &x2, const NodePtr &bias, const NodePtr &scales, const NodePtr &expanded_row_idx, const NodePtr &expanded_expert_idx);
     static NodePtr MoeInitRouting(const NodePtr &x, const NodePtr &row_idx, const NodePtr &expert_idx, const NodePtr &active_num);
-    static NodePtr AddRmsNormQuantV2(const NodePtr &x1, const NodePtr &x2, const NodePtr &gamma, const NodePtr &scale, const NodePtr &offset, const NodePtr &epsilon);
+    static NodePtr GroupedMatmulV4(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &scale, const NodePtr &offset, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &pre_token_scale, const NodePtr &group_list, const NodePtr &activation_input, const NodePtr &activation_quant_scale, const NodePtr &activation_quant_offset, const NodePtr &split_item, const NodePtr &group_type, const NodePtr &group_list_type, const NodePtr &act_type);
     static NodePtr DynamicQuantExt(const NodePtr &x, const NodePtr &smooth_scales);
+    static NodePtr QuantMatmul(const NodePtr &x1, const NodePtr &x2, const NodePtr &scale, const NodePtr &offset, const NodePtr &pertoken_scale, const NodePtr &bias, const NodePtr &output_dtype, const NodePtr &x1_dtype, const NodePtr &x2_dtype, const NodePtr &pertoken_scale_dtype, const NodePtr &scale_dtype, const NodePtr &group_sizes);
     static NodePtr GroupedMatmulV2(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &scale, const NodePtr &offset, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &group_list, const NodePtr &split_item, const NodePtr &group_type);
     static NodePtr MoeComputeExpertTokens(const NodePtr &sorted_experts, const NodePtr &num_expert);
-    static NodePtr GroupedMatmul(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &scale, const NodePtr &offset, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &group_list, const NodePtr &split_item, const NodePtr &group_type, const NodePtr &transpose_a, const NodePtr &transpose_b);
-    static NodePtr GroupedMatmulV4(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &scale, const NodePtr &offset, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &pre_token_scale, const NodePtr &group_list, const NodePtr &activation_input, const NodePtr &activation_quant_scale, const NodePtr &activation_quant_offset, const NodePtr &split_item, const NodePtr &group_type, const NodePtr &group_list_type, const NodePtr &act_type);
     static NodePtr WeightQuantBatchMatmul(const NodePtr &x, const NodePtr &weight, const NodePtr &antiquant_scale, const NodePtr &antiquant_offset, const NodePtr &quant_scale, const NodePtr &quant_offset, const NodePtr &bias, const NodePtr &transpose_x, const NodePtr &transpose_weight, const NodePtr &antiquant_group_size);
-    static NodePtr MoeInitRoutingV2(const NodePtr &x, const NodePtr &expert_idx, const NodePtr &active_num, const NodePtr &expert_capacity, const NodePtr &expert_num, const NodePtr &drop_pad_mode, const NodePtr &expert_tokens_count_or_cumsum_flag, const NodePtr &expert_tokens_before_capacity_flag);
+    static NodePtr QuantBatchMatmul(const NodePtr &x1, const NodePtr &x2, const NodePtr &scale, const NodePtr &offset, const NodePtr &bias, const NodePtr &pertokenScaleOptional, const NodePtr &transpose_x1, const NodePtr &transpose_x2, const NodePtr &dtype);
+    static NodePtr GmmBackward(const NodePtr &grad, const NodePtr &x, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);
+    static NodePtr EinsumExt(const NodePtr &equation, const NodePtr &operands);
+    static NodePtr GmmBackwardFusion(const NodePtr &grad, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);
+    static NodePtr GmmV2(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &group_list, const NodePtr &group_type, const NodePtr &group_list_type);
+    static NodePtr AnyExt(const NodePtr &input, const NodePtr &dim, const NodePtr &keepdim);
+    static NodePtr Any(const NodePtr &input);
     static NodePtr GmmV2Backward(const NodePtr &grad, const NodePtr &x, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);
     static NodePtr Gmm(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &group_list, const NodePtr &group_type, const NodePtr &group_list_type);
+    static NodePtr MoeTokenUnpermute(const NodePtr &permuted_tokens, const NodePtr &sorted_indices, const NodePtr &probs, const NodePtr &padded_mode, const NodePtr &restore_shape);
     static NodePtr InplaceExponential(const NodePtr &input, const NodePtr &lambd, const NodePtr &seed, const NodePtr &offset);
-    static NodePtr GmmV2(const NodePtr &x, const NodePtr &weight, const NodePtr &bias, const NodePtr &group_list, const NodePtr &group_type, const NodePtr &group_list_type);
-    static NodePtr GmmBackward(const NodePtr &grad, const NodePtr &x, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);
+    static NodePtr FuncMaxPool2D(const NodePtr &input, const NodePtr &kernel_size, const NodePtr &stride, const NodePtr &padding, const NodePtr &dilation, const NodePtr &ceil_mode, const NodePtr &return_indices);
+    static NodePtr FuncDropoutExt(const NodePtr &input, const NodePtr &p, const NodePtr &training, const NodePtr &inplace, const NodePtr &seed, const NodePtr &offset);
     static NodePtr GmmV2BackwardFusion(const NodePtr &grad, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);
-    static NodePtr PixelShuffle(const NodePtr &input, const NodePtr &upscale_factor);
-    static NodePtr GmmBackwardFusion(const NodePtr &grad, const NodePtr &weight, const NodePtr &group_list, const NodePtr &group_list_type);;
+    static NodePtr PixelShuffle(const NodePtr &input, const NodePtr &upscale_factor);;
   private:
     static std::string device_target_;
 };