Enzyme: mark device_sync as non-differentiable (#2605)

[only downstream]
JuliaGPU · Dec 27, 2024 · 792aec5 · 792aec5
1 parent a0c2f4b
commit 792aec5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ext/EnzymeCoreExt.jl b/ext/EnzymeCoreExt.jl
@@ -25,6 +25,9 @@ end
 function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(CUDA.is_pinned), args...)
     return nothing
 end
+function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(CUDA.device_synchronize), args...)
+    return nothing
+end
 function EnzymeCore.EnzymeRules.inactive(::typeof(CUDA.launch_configuration), args...; kwargs...)
     return nothing
 end