executor error handling for gpu cases

youwuyou · Jun 2, 2024 · 4ec0a30 · 4ec0a30
1 parent fff44d5
commit 4ec0a30
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/base/Executor.jl b/src/base/Executor.jl
@@ -71,7 +71,13 @@ function create(executor_type::Symbol; device_id::Integer = 0)
     if executor_type == :reference || executor_type == :omp
         return GkoExecutor(executor_type)
     else
-        return GkoExecutor(executor_type, device_id = device_id)
+        # on GPU
+        num_device = get_num_devices(executor_type)
+        if num_device > 0
+            return GkoExecutor(executor_type, device_id = device_id)
+        else
+            throw(ArgumentError("No device available for $executor_type, num_device = $num_device"))
+        end
     end
 end