From fe2f491dd711d50ea91da6064411f594bf994e91 Mon Sep 17 00:00:00 2001
From: Adnan Akhundov <adnan.akhundov@gmail.com>
Date: Fri, 28 Apr 2023 19:23:23 +0200
Subject: [PATCH] Get SM count with cudaDeviceGetAttribute in
 KernelHardwareInfo (#927)

---
 include/cutlass/kernel_hardware_info.hpp | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/include/cutlass/kernel_hardware_info.hpp b/include/cutlass/kernel_hardware_info.hpp
index 3ae09324..586ef723 100644
--- a/include/cutlass/kernel_hardware_info.hpp
+++ b/include/cutlass/kernel_hardware_info.hpp
@@ -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;
   }
 };