昇腾社区首页
中文
注册

接口列表

算子归属领域

算子接口可被归属为如下领域:

  • aclnn_ops_infer:NN网络算子推理库
  • aclnn_ops_train:NN网络算子训练库
  • aclnn_math:数学算子库
  • aclnn_rand:随机数算子库

其中推理库依赖数学库,训练库依赖推理库、数学库与随机库。

每个算子接口所属领域如下表所示。

aclnn api aclnn_ops_infer aclnn_ops_train aclnn_math aclnn_rand
aclnnAddLayerNorm
aclnnAddLayerNormGrad
aclnnAddRmsNorm
aclnnAllGatherMatmul
aclnnAlltoAllAllGatherBatchMatMul
aclnnApplyRotaryPosEmb
aclnnBidirectionLSTM
aclnnBatchMatmulReduceScatterAlltoAll
aclnnDeepNorm
aclnnDeepNormGrad
aclnnDynamicQuant
aclnnEinsum
aclnnFFN(代码开放)
aclnnFFNV2(代码开放)
aclnnFFNV3(代码开放)
aclnnFlashAttentionScore(代码开放)
aclnnFlashAttentionScoreGrad(代码开放)
aclnnFlashAttentionScoreV2(代码开放)
aclnnFlashAttentionScoreGradV2(代码开放)
aclnnFlashAttentionVarLenScore(代码开放)
aclnnFlashAttentionVarLenScoreV2(代码开放)
aclnnFlashAttentionUnpaddingScoreGrad(代码开放)
aclnnFlashAttentionUnpaddingScoreGradV2(代码开放)
aclnnFusedInferAttentionScore(代码开放)
aclnnFusedInferAttentionScoreV2(代码开放)
aclnnGroupedMatmul
aclnnGroupedMatMulAllReduce
aclnnGroupedMatmulV2
aclnnGroupedMatmulV3
aclnnGroupedMatmulV4
aclnnIncreFlashAttention(代码开放)
aclnnIncreFlashAttentionV2(代码开放)
aclnnIncreFlashAttentionV3(代码开放)
aclnnIncreFlashAttentionV4(代码开放)
aclnnInplaceQuantScatter
aclnnInstanceNorm
aclnnMatmulAllReduce
aclnnMatmulAllReduceAddRmsNorm
aclnnMatmulAllReduceV2
aclnnMatmulReduceScatter
aclnnMoeFinalizeRouting
aclnnMoeGatingTopKSoftmax
aclnnMoeInitRouting
aclnnMoeTokenPermute
aclnnMoeTokenPermuteGrad
aclnnMoeTokenUnpermute
aclnnMoeTokenUnpermuteGrad
aclnnMrgbaCustom
aclnnMultiScaleDeformableAttentionGrad
aclnnPromptFlashAttention(代码开放)
aclnnPromptFlashAttentionV2(代码开放)
aclnnPromptFlashAttentionV3(代码开放)
aclnnQuantMatmulAllReduce
aclnnQuantMatmulAllReduceV2
aclnnQuantMatmulAllReduceV3
aclnnQuantMatmulAllReduceAddRmsNorm
aclnnRmsNorm
aclnnRmsNormGrad
aclnnRingAttentionUpdate
aclnnSwinTransformerLnQkvQuant
aclnnSwinAttentionScoreQuant
aclnnWeightQuantBatchMatmul
aclnnWeightQuantBatchMatmulV2
aclnnWeightQuantMatmulAllReduce
aclnnWeightQuantMatmulAllReduceAddRmsNorm
aclRfft1D
aclStft

废弃接口

废弃接口 说明
aclnnWeightQuantBatchMatmul 此接口后续版本会废弃,请使用 aclnnWeightQuantBatchMatmulV2接口。