tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1diff --git a/src/init.cc b/src/init.cc
2index 738f756..1b0e4fc 100644
3--- a/src/init.cc
4+++ b/src/init.cc
5@@ -2049,7 +2049,7 @@ static ncclResult_t ncclCommInitRankFunc(struct ncclAsyncJob* job_) {
6 if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled()) && mscclppCommCompatible(comm)) {
7 hipDeviceProp_t devProp;
8 CUDACHECK(hipGetDeviceProperties(&devProp, cudaDev));
9- comm->mscclppCompatible = IsArchMatch(devProp.gcnArchName, "gfx94");
10+ comm->mscclppCompatible = IsArchMatch(devProp.gcnArchName, "gfx9");
11 if (comm->mscclppCompatible) {
12 bool mapContainsId = (mscclpp_uniqueIdMap.count(job->commId) > 0);
13 auto& mscclppUniqueId = mscclpp_uniqueIdMap[job->commId];