Disable GPU methods if no device available (#1213)

Qiskit · Apr 14, 2021 · 1d95e90 · 1d95e90
1 parent 89b5a1b
commit 1d95e90
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/src/controllers/aer_controller.hpp b/src/controllers/aer_controller.hpp
@@ -502,6 +502,12 @@ void Controller::set_config(const json_t &config) {
         throw std::runtime_error(
             "Simulation device \"GPU\" is not supported on this system");
 #else
+        int nDev;
+        if (cudaGetDeviceCount(&nDev) != cudaSuccess) {
+            cudaGetLastError();
+            throw std::runtime_error("No CUDA device available!");
+        }
+
         sim_device_ = Device::GPU;
 #endif
       }

diff --git a/src/controllers/controller.hpp b/src/controllers/controller.hpp
@@ -300,6 +300,20 @@ class Controller {
 //-------------------------------------------------------------------------
 
 void Controller::set_config(const json_t &config) {
+#ifdef AER_THRUST_CUDA
+  {
+    std::string method;
+    if (JSON::get_value(method, "method", config)) {
+      if(method.find("gpu") != std::string::npos){
+        int nDev;
+        if (cudaGetDeviceCount(&nDev) != cudaSuccess) {
+          cudaGetLastError();
+          throw std::runtime_error("No CUDA device available!");
+        }
+      }
+    }
+  }
+#endif
 
   // Load validation threshold
   JSON::get_value(validation_threshold_, "validation_threshold", config);