Skip to content

Commit 792aec5

Browse files
authored
Enzyme: mark device_sync as non-differentiable (#2605)
[only downstream]
1 parent a0c2f4b commit 792aec5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

ext/EnzymeCoreExt.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,9 @@ end
2525
function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(CUDA.is_pinned), args...)
2626
return nothing
2727
end
28+
function EnzymeCore.EnzymeRules.inactive_noinl(::typeof(CUDA.device_synchronize), args...)
29+
return nothing
30+
end
2831
function EnzymeCore.EnzymeRules.inactive(::typeof(CUDA.launch_configuration), args...; kwargs...)
2932
return nothing
3033
end

0 commit comments

Comments
 (0)