Add Enzyme GPU support

Fix examples tests with CUDA backend Add synchronize rule
JuliaGPU · May 31, 2024 · c21f6bb · c21f6bb
1 parent bf25c71
commit c21f6bb
Show file tree

Hide file tree

Showing 10 changed files with 379 additions and 122 deletions.
diff --git a/Project.toml b/Project.toml
@@ -21,7 +21,7 @@ UnsafeAtomicsLLVM = "d80eeb9a-aca5-4d75-85e5-170c8b632249"
 [compat]
 Adapt = "0.4, 1.0, 2.0, 3.0, 4"
 Atomix = "0.1"
-EnzymeCore = "0.7.1"
+EnzymeCore = "0.7.2"
 InteractiveUtils = "1.6"
 LinearAlgebra = "1.6"
 MacroTools = "0.5"

diff --git a/examples/utils.jl b/examples/utils.jl
@@ -1,5 +1,5 @@
 # EXCLUDE FROM TESTING
-if Base.find_package("CUDA") !== nothing
+if backend_str == "CUDA" && Base.find_package("CUDA") !== nothing
     using CUDA
     using CUDA.CUDAKernels
     const backend = CUDABackend()