microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-11 17:00:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updating NpKit (#785)

Browse Source
This commit is contained in:
Caio Rocha
2026-04-13 13:36:42 -07:00
committed by GitHub
parent 5380a4ac6e
commit b59e6d7f00
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/core/npkit/npkit.cc
View File

@@ -103,10 +103,10 @@ static int GetGpuClockRateInKhz() {
else
return 25000;
#else
cudaDeviceProp dev_prop;
int clockRate;
MSCCLPP_CUDATHROW(cudaGetDevice(&dev_id));
MSCCLPP_CUDATHROW(cudaGetDeviceProperties(&dev_prop, dev_id));
return dev_prop.clockRate;
MSCCLPP_CUDATHROW(cudaDeviceGetAttribute(&clockRate, cudaDevAttrClockRate, dev_id));
return clockRate;
#endif
}
#endif