Fix crash when NPKit is enabled (#887)

[ROCm/rccl commit: 6a4d5ec089]
This commit is contained in:
Wenkai Du
2023-09-13 11:00:12 -07:00
committed by GitHub
parent 2e3d45a53a
commit b0a16d80ff
+1 -1
View File
@@ -72,7 +72,7 @@ int GetGcnArchName(int deviceId, char* out) {
}
double GetDeviceWallClockRateInKhz(int deviceId) {
char* gcn;
char gcn[256];
GetGcnArchName(deviceId, gcn);
if (strncmp("gfx94", gcn, 5) == 0)
return 1.0E5;