已合并
change xn for alltoall in ccu scene according to hccl #1132
zhangouli创建于 3月23日
change xn for alltoall in ccu scene according to hccl #1132
已合并
共 2 个文件变更+3-0
| @@ -52,6 +52,8 @@ HcclImpl<HcclServerType::HCCL_SERVER_TYPE_CCU, config>::CcuPrepareForAllToAll(__ | |||
| 52 | xnData_[4] = strideSize; | 52 | xnData_[4] = strideSize; |
| 53 | xnData_[5] = 0; | 53 | xnData_[5] = 0; |
| 54 | xnData_[6] = strideSize * ccuParam_.rankId; | 54 | xnData_[6] = strideSize * ccuParam_.rankId; |
| 55 | uint64_t loopCount = 8; | ||
| 56 | CalcGoSize(sliceSizeAlltoall, loopCount, CCU_MEMSLICE_SIZE * 8, &xnData_[7]); | ||
| 55 | return; | 57 | return; |
| 56 | } | 58 | } |
| 57 | 59 | ||
| @@ -241,6 +241,7 @@ HcclImpl<HcclServerType::HCCL_SERVER_TYPE_CCU, config>::CcuPrepareForOp(const Hc | |||
| 241 | CcuPrepareForAllReduce(&handleParamGM_[handleId]); | 241 | CcuPrepareForAllReduce(&handleParamGM_[handleId]); |
| 242 | } | 242 | } |
| 243 | } else if (handleParamGM_[handleId].commType.prepareType == HcclCMDType::HCCL_CMD_ALLTOALL) { | 243 | } else if (handleParamGM_[handleId].commType.prepareType == HcclCMDType::HCCL_CMD_ALLTOALL) { |
| 244 | ccuUsedXnNum_ = 11; | ||
| 244 | CcuPrepareForAllToAll(&handleParamGM_[handleId]); | 245 | CcuPrepareForAllToAll(&handleParamGM_[handleId]); |
| 245 | } else if (handleParamGM_[handleId].commType.prepareType == HcclCMDType::HCCL_CMD_ALLTOALLV) { | 246 | } else if (handleParamGM_[handleId].commType.prepareType == HcclCMDType::HCCL_CMD_ALLTOALLV) { |
| 246 | FlushDataCache(&allToAllVParam_[handleId]); | 247 | FlushDataCache(&allToAllVParam_[handleId]); |