Reinstated the code that ensured cudaGetDeviceProperties is not repea…

…tedly called. Signed-off-by: Marat Arsaev <marsaev@nvidia.com>
NVIDIA · Apr 7, 2020 · 83629fe · 83629fe
1 parent cb986b3
commit 83629fe
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/base/src/device_properties.cu b/base/src/device_properties.cu
@@ -28,16 +28,16 @@
 namespace amgx
 {
 static cudaDeviceProp deviceProps;
-//static bool initialized=false;
+static bool initialized=false;
 
 cudaDeviceProp getDeviceProperties()
 {
-    //if(!initialized) {
-    int dev;
-    cudaGetDevice(&dev);
-    cudaGetDeviceProperties(&deviceProps, dev);
-    //initialized=true;
-    //}
+    if(!initialized) {
+        int dev;
+        cudaGetDevice(&dev);
+        cudaGetDeviceProperties(&deviceProps, dev);
+        initialized=true;
+    }
     return deviceProps;
 }