Skip to content

Commit

Permalink
Update compiler.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
wsmoses authored Dec 7, 2024
1 parent ea1c69b commit 159d33b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/compiler.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5558,13 +5558,13 @@ function thunk_generator(world::UInt, source::LineNumberNode, @nospecialize(FA::
edges = Core.MethodInstance[mi]

if Mode == API.DEM_ForwardMode
push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{typeof(EnzymeRules.forward)}, world)
push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{typeof(EnzymeRules.forward)}, world))
Compiler.Interpreter.rule_backedge_holder(Base.inferencebarrier(EnzymeRules.forward))
else
push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{typeof(EnzymeRules.augmented_forward)}, world)
push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{typeof(EnzymeRules.augmented_forward)}, world))
end

push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{Val{0}}, world)
push!(edges, GPUCompiler.methodinstance(typeof(Compiler.Interpreter.rule_backedge_holder), Tuple{Val{0}}, world))
Compiler.Interpreter.rule_backedge_holder(Base.inferencebarrier(Val(0)))

new_ci.edges = edges
Expand Down

0 comments on commit 159d33b

Please sign in to comment.