diff --git a/src/include/gdrwrap.h b/src/include/gdrwrap.h index a71dddb54a..1954f00d1d 100644 --- a/src/include/gdrwrap.h +++ b/src/include/gdrwrap.h @@ -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 {