Skip to content

Commit c21f6bb

Browse files
committed
Add Enzyme GPU support
Fix examples tests with CUDA backend Add synchronize rule
1 parent bf25c71 commit c21f6bb

File tree

10 files changed

+379
-122
lines changed

10 files changed

+379
-122
lines changed

Diff for: Project.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ UnsafeAtomicsLLVM = "d80eeb9a-aca5-4d75-85e5-170c8b632249"
2121
[compat]
2222
Adapt = "0.4, 1.0, 2.0, 3.0, 4"
2323
Atomix = "0.1"
24-
EnzymeCore = "0.7.1"
24+
EnzymeCore = "0.7.2"
2525
InteractiveUtils = "1.6"
2626
LinearAlgebra = "1.6"
2727
MacroTools = "0.5"

Diff for: examples/utils.jl

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# EXCLUDE FROM TESTING
2-
if Base.find_package("CUDA") !== nothing
2+
if backend_str == "CUDA" && Base.find_package("CUDA") !== nothing
33
using CUDA
44
using CUDA.CUDAKernels
55
const backend = CUDABackend()

0 commit comments

Comments
 (0)