Skip to content

Commit 6169d9c

Browse files
committed
Avoid blow up of function pre-amble on higer order AD
1 parent 4656760 commit 6169d9c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/compiler.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1327,6 +1327,9 @@ struct PrimalCompilerParams <: AbstractEnzymeCompilerParams
13271327
mode::API.CDerivativeMode
13281328
end
13291329

1330+
# Avoid blow-up of higer-order AD
1331+
GPUCompiler.can_safepoint(::CompilerJob{<:Any,<:AbstractEnzymeCompilerParams}) = false
1332+
13301333
DefaultCompilerTarget(; kwargs...) =
13311334
GPUCompiler.NativeCompilerTarget(; jlruntime = true, kwargs...)
13321335

0 commit comments

Comments
 (0)