From cc4de02a86a3084207ebd432bd21bf1efab185a0 Mon Sep 17 00:00:00 2001 From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com> Date: Mon, 23 Oct 2023 12:04:37 -0700 Subject: [PATCH] Add missing gfx942 support (#927) [ROCm/rccl commit: c4e65fd38270aad2f01879d2d53235864038d77c] --- projects/rccl/src/graph/topo.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/projects/rccl/src/graph/topo.cc b/projects/rccl/src/graph/topo.cc index cf8913b7a3..b7b820a1e0 100644 --- a/projects/rccl/src/graph/topo.cc +++ b/projects/rccl/src/graph/topo.cc @@ -380,7 +380,10 @@ ncclResult_t ncclTopoAddGpu(struct ncclXmlNode* xmlGpu, struct ncclTopoSystem* s gpu->gpu.gcn = "gfx940"; } else if (strcmp(gpu->gpu.gcn, "941") == 0) { gpu->gpu.gcn = "gfx941"; + } else if (strcmp(gpu->gpu.gcn, "942") == 0) { + gpu->gpu.gcn = "gfx942"; } + rcclHipDeviceArch_t arch; NCCLCHECK(xmlGetAttrInt(xmlGpu, "arch", &arch.value)); memcpy(&gpu->gpu.arch, &arch.arch, sizeof(hipDeviceArch_t));