Use cudaDeviceGetAttribute instead of cudaGetDeviceProperties (#5570)

2020-04-21 14:58:29 +12:00
parent 73142041b9
commit a734f52807
1 changed files with 5 additions and 3 deletions
--- a/src/common/device_helpers.cuh
+++ b/src/common/device_helpers.cuh
@@ -108,9 +108,11 @@ inline size_t TotalMemory(int device_idx) {
 */

 inline size_t MaxSharedMemory(int device_idx) {
-  cudaDeviceProp prop;
-  dh::safe_cuda(cudaGetDeviceProperties(&prop, device_idx));
-  return prop.sharedMemPerBlock;
+  int max_shared_memory = 0;
+  dh::safe_cuda(cudaDeviceGetAttribute
+                (&max_shared_memory, cudaDevAttrMaxSharedMemoryPerBlock,
+                 device_idx));
+  return size_t(max_shared_memory);
 }

 /**