已合并
fix allreduce err #4672
hu-yiliang11创建于 8月18日
fix allreduce err #4672
已合并
共 1 个文件变更+2-1
| @@ -639,7 +639,8 @@ HcclResult AllReduceOperator::SelectAlgfor91093(const OpParam& param, std::strin | |||
| 639 | && ((topoMatcher_->GetDeterministicConfig() != DETERMINISTIC_DISABLE) || (serverNum_ > 1)) | 639 | && ((topoMatcher_->GetDeterministicConfig() != DETERMINISTIC_DISABLE) || (serverNum_ > 1)) |
| 640 | && ((userRankSize_ > DEVICE_EIGHT && dataSize < HCCL_SMALL_COUNT_8_MB) | 640 | && ((userRankSize_ > DEVICE_EIGHT && dataSize < HCCL_SMALL_COUNT_8_MB) |
| 641 | || (userRankSize_ <= DEVICE_EIGHT && dataSize <= HCCL_SMALL_COUNT_512_KB) || isOnlyAiv) | 641 | || (userRankSize_ <= DEVICE_EIGHT && dataSize <= HCCL_SMALL_COUNT_512_KB) || isOnlyAiv) |
| 642 | - && (!retryEnable_) && userRankSize_ > 1 && !multiModuleDiffDeviceNumMode_; | 642 | + && (!retryEnable_) && userRankSize_ > 1 && !multiModuleDiffDeviceNumMode_ |
| 643 | + && ((!GetExternalInputInterHccsDisable()) || (serverNum_ == 1)); | ||
| 643 | 644 | ||
| 644 | bool isAivMode = (topoMatcher_->GetAivModeConfig() && !isBarrierOp) | 645 | bool isAivMode = (topoMatcher_->GetAivModeConfig() && !isBarrierOp) |
| 645 | && IsSupportAIVReduce(param.DataDes.dataType, param.reduceType) && serverNum_ == 1 | 646 | && IsSupportAIVReduce(param.DataDes.dataType, param.reduceType) && serverNum_ == 1 |