enable building rccl for gfx950 (#1571)

[ROCm/rccl commit: f268553ee4]
This commit is contained in:
Pedram Alizadeh
2025-02-25 16:13:48 -05:00
committed by GitHub
parent c363118589
commit acf5822a6c
24 changed files with 66 additions and 47 deletions
+4
View File
@@ -48,6 +48,8 @@ void convertGcnArchToGcnArchName(const char* gcnArch, const char** gcnArchName)
*gcnArchName = "gfx941";
else if (strcmp(gcnArch, "942") == 0)
*gcnArchName = "gfx942";
else if (strcmp(gcnArch, "950") == 0)
*gcnArchName = "gfx950";
else
*gcnArchName = gcnArch;
}
@@ -65,6 +67,8 @@ double GetDeviceWallClockRateInKhz(int deviceId) {
GetGcnArchName(deviceId, gcn);
if (strncmp("gfx94", gcn, 5) == 0)
return 1.0E5;
else if(strncmp("gfx950", gcn, 6) == 0)
return 1.0E5;
else
return 2.5E4;
}