Enabling gdrcopy option for gfx950 (#1955)

2025-10-15 10:55:25 -06:00
parent c70f5b4621
commit fedddb452c
1 changed files with 2 additions and 1 deletions
@@ -169,7 +169,8 @@ static gdr_t ncclGdrInit() {
    return NULL;
  }
  GcnArchNameFormat(devProp.gcnArchName, gcnArchNameSubstr);
-  if (IsArchMatch(gcnArchNameSubstr, "gfx942")) {
+  if (IsArchMatch(gcnArchNameSubstr, "gfx942") ||
+      IsArchMatch(gcnArchNameSubstr, "gfx950")) {
    INFO(NCCL_INIT, "Enabled GDRCopy equivalent memory allocation on %s", gcnArchNameSubstr);
    return (gdr_t)0x12345678L;
  } else {