compile

Author: zzx-study

csrc/deepep/ops2/op_host/op_api/aclnn_dispatch_normal_a2.cpp

@@ -17,7 +17,7 @@ extern "C" {
 #endif
 
 aclnnStatus aclnnDispatchNormalA2GetWorkspaceSize(
-    const aclTensor *x, const aclTensor *expertIds, const aclTensor *tokenPerRank, const aclTensor *scales, const aclTensor *xActiveMask,
+    const aclTensor *x, const aclTensor *expertIds, const aclTensor *scales, const aclTensor *xActiveMask,
     const aclTensor *expertScales, const aclTensor *tokenServerIdx, const aclTensor *tokenServerCnt,
     const aclTensor *epRankTokenCnt, const aclTensor *srcOffsetRankTokenIdx, const aclTensor *dstOffsetRankTokenIdx, const aclTensor *tokenIdxPerExpert,
     char *groupEp, int64_t epWorldSize, int64_t epRankId, int64_t moeExpertNum, char *groupTp, int64_t tpWorldSize,
@@ -28,7 +28,7 @@ aclnnStatus aclnnDispatchNormalA2GetWorkspaceSize(
     aclOpExecutor **executor)
 {
     return aclnnInnerDispatchNormalA2GetWorkspaceSize(
-        x, expertIds, tokenPerRank, scales, xActiveMask, expertScales, tokenServerIdx, tokenServerCnt, epRankTokenCnt,
+        x, expertIds, scales, xActiveMask, expertScales, tokenServerIdx, tokenServerCnt, epRankTokenCnt,
         srcOffsetRankTokenIdx, dstOffsetRankTokenIdx,tokenIdxPerExpert, groupEp, epWorldSize, epRankId, moeExpertNum, groupTp,
         tpWorldSize, tpRankId, expertShardType, sharedExpertNum, sharedExpertRankNum, quantMode, globalBs,
         expertTokenNumsType, recvX, dynamicScales, expandIdx, expertTokenNums, epRecvCount, expandScales,