youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Get SM count with cudaDeviceGetAttribute in KernelHardwareInfo (#927)

Browse Source
This commit is contained in:
Adnan Akhundov
2023-04-28 19:23:23 +02:00
committed by GitHub
parent df02482f1d
commit fe2f491dd7
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/cutlass/kernel_hardware_info.hpp
View File

@ -56,15 +56,16 @@ struct KernelHardwareInfo {
<< cudaGetErrorString(result));
return 0;
}
cudaDeviceProp properties;
result = cudaGetDeviceProperties(&properties, device_id);
int multiprocessor_count;
result = cudaDeviceGetAttribute(&multiprocessor_count,
cudaDevAttrMultiProcessorCount, device_id);
if (result != cudaSuccess) {
CUTLASS_TRACE_HOST(
" cudaGetDeviceProperties() returned error "
" cudaDeviceGetAttribute() returned error "
<< cudaGetErrorString(result));
return 0;
}
return properties.multiProcessorCount;
return multiprocessor_count;
}
};