Merge pull request #370 from toxa81/develop

minor simplification
electronic-structure · May 13, 2019 · b118821 · b118821
2 parents 19dcb9c + a6aff48
commit b118821
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/SDDK/GPU/acc.hpp b/src/SDDK/GPU/acc.hpp
@@ -180,17 +180,14 @@ namespace acc {
 /// Get the number of devices.
 inline int num_devices()
 {
-    static int count{-1};
 #if defined(__CUDA) || defined(__ROCM)
+    static int count{-1};
     if (count == -1) {
         if (GPU_PREFIX(GetDeviceCount)(&count) != GPU_PREFIX(Success)) {
-            return 0;
-        } else {
-            return count;
+            count = 0;
         }
-    } else {
-        return count;
     }
+    return count;
 #else
     return 0;
 #endif