Merge pull request #26 from TensorBFS/0.1.1

Add device detector
TensorBFS · Oct 19, 2023 · dfa895f · dfa895f · ArrogantGao · Oct 19, 2023
2 parents d0066a4 + 62f24b1
commit dfa895f
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/CuTropicalGEMM.jl b/src/CuTropicalGEMM.jl
@@ -4,8 +4,15 @@ using CUDA, TropicalNumbers, LinearAlgebra, TropicalGemmC_jll
 export matmul!
 
 function __init__()
-    @assert CUDA.driver_version() >= v"11.4" "Error: CUDA.driver_version < v11.4"
-    @assert CUDA.driver_version() <= v"12.2" "Error: CUDA.driver_version > v12.2"
+    if CUDA.functional() == true
+        if CUDA.driver_version() < v"11.4"
+            @warn "CUDA.driver_version < v11.4! CuTropicalGEMM may not be available."
+        elseif CUDA.driver_version() > v"12.2"
+            @warn "CUDA.driver_version > v12.2! CuTropicalGEMM may not be available."
+        end
+    elseif CUDA.functional() == false
+        @warn "CUDA Driver not found! CuTropicalGEMM will not be available."
+    end
     return nothing
 end