squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats 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;
}
};