14 lines
733 B
Diff
14 lines
733 B
Diff
|
diff --git a/src/init.cc b/src/init.cc
|
||
|
|
index 738f756..1b0e4fc 100644
|
||
|
|
--- a/src/init.cc
|
||
|
|
+++ b/src/init.cc
|
||
|
|
@@ -2049,7 +2049,7 @@ static ncclResult_t ncclCommInitRankFunc(struct ncclAsyncJob* job_) {
|
||
|
|
if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled()) && mscclppCommCompatible(comm)) {
|
||
|
|
hipDeviceProp_t devProp;
|
||
|
|
CUDACHECK(hipGetDeviceProperties(&devProp, cudaDev));
|
||
|
|
- comm->mscclppCompatible = IsArchMatch(devProp.gcnArchName, "gfx94");
|
||
|
|
+ comm->mscclppCompatible = IsArchMatch(devProp.gcnArchName, "gfx9");
|
||
|
|
if (comm->mscclppCompatible) {
|
||
|
|
bool mapContainsId = (mscclpp_uniqueIdMap.count(job->commId) > 0);
|
||
|
|
auto& mscclppUniqueId = mscclpp_uniqueIdMap[job->commId];
|