Enzyme: mark device_sync as non-differentiable [only downstream]

leads to a compile-time performance boost
JuliaGPU · Dec 26, 2024 · 09db71a · 09db71a
1 parent a0c2f4b
commit 09db71a
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