下载
EN
注册

接口列表

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

  • 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
aclnnApplyRotaryPosEmb
aclnnBidirectionLSTM
aclnnDeepNorm
aclnnDeepNormGrad
aclnnDynamicQuant
aclnnDynamicQuantV2
aclnnEinsum
aclnnFFN(代码开放)
aclnnFFNV2(代码开放)
aclnnFFNV3(代码开放)
aclnnFlashAttentionScore(代码开放)
aclnnFlashAttentionScoreV2(代码开放)
aclnnFlashAttentionScoreGrad(代码开放)
aclnnFlashAttentionScoreGradV2(代码开放)
aclnnFlashAttentionVarLenScore(代码开放)
aclnnFlashAttentionVarLenScoreV2(代码开放)
aclnnFlashAttentionUnpaddingScoreGrad(代码开放)
aclnnFlashAttentionUnpaddingScoreGradV2(代码开放)
aclnnFusedInferAttentionScore(代码开放)
aclnnFusedInferAttentionScoreV2(代码开放)
aclnnGroupedMatmul
aclnnGroupedMatmulV2
aclnnGroupedMatmulV3
aclnnGroupedMatmulV4
aclnnGroupedMatMulAllReduce
aclnnIncreFlashAttention(代码开放)
aclnnIncreFlashAttentionV2(代码开放)
aclnnIncreFlashAttentionV3(代码开放)
aclnnIncreFlashAttentionV4(代码开放)
aclnnInplaceMatmulAllReduceAddRmsNorm
aclnnInplaceQuantMatmulAllReduceAddRmsNorm
aclnnInplaceQuantScatter
aclnnInplaceWeightQuantMatmulAllReduceAddRmsNorm
aclnnInstanceNorm
aclnnMatmulAllReduce
aclnnMatmulAllReduceV2
aclnnMatmulAllReduceAddRmsNorm
aclnnMatmulReduceScatter
aclnnMoeComputeExpertTokens
aclnnMoeFinalizeRouting
aclnnMoeFinalizeRoutingV2
aclnnMoeGatingTopKSoftmax
aclnnMoeInitRouting
aclnnMoeInitRoutingQuant
aclnnMoeTokenPermute
aclnnMoeTokenPermuteGrad
aclnnMoeTokenUnpermute
aclnnMoeTokenUnpermuteGrad
aclnnMrgbaCustom
aclnnMultiScaleDeformableAttentionGrad
aclnnPromptFlashAttention(代码开放)
aclnnPromptFlashAttentionV2(代码开放)
aclnnPromptFlashAttentionV3(代码开放)
aclnnQuantMatmulAllReduce
aclnnQuantMatmulAllReduceV2
aclnnQuantMatmulAllReduceV3
aclnnQuantMatmulAllReduceAddRmsNorm
aclnnRingAttentionUpdate
aclnnRmsNorm
aclnnRmsNormGrad
aclnnSwinAttentionScoreQuant
aclnnSwinTransformerLnQkvQuant
aclnnWeightQuantBatchMatmulV2
aclnnWeightQuantMatmulAllReduce
aclnnWeightQuantMatmulAllReduceAddRmsNorm
aclRfft1D
aclStft