fix

EnzymeAD · Apr 23, 2023 · 9178424 · 9178424
1 parent e294f0d
commit 9178424
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -6245,7 +6245,7 @@ struct EnzymeCompilerParams <: AbstractEnzymeCompilerParams
     expectedTapeType::Type
     method_table
 end
-GPUCompiler.method_table(@nospecialize(job::CompilerJob{T,EnzymeCompilerParams})) where T = job.config.params.method_table
+GPUCompiler.method_table(@nospecialize(job::CompilerJob{T,EnzymeCompilerParams})) where T = job.config.params.method_table !== nothing ? job.config.params.method_table : GPUCompiler.GLOBAL_METHOD_TABLE
 
 struct UnknownTapeType end
 

diff --git a/src/compiler/reflection.jl b/src/compiler/reflection.jl
@@ -1,5 +1,5 @@
 function get_job(@nospecialize(func), @nospecialize(A), @nospecialize(types);
-        run_enzyme::Bool=true, mode::API.CDerivativeMode=API.DEM_ReverseModeCombined, dupClosure::Bool=false, argwrap::Bool=true, width::Int=1, modifiedBetween=nothing, returnPrimal::Bool=false, augmentedInit=false, world=nothing, kwargs...)
+        run_enzyme::Bool=true, mode::API.CDerivativeMode=API.DEM_ReverseModeCombined, dupClosure::Bool=false, argwrap::Bool=true, width::Int=1, modifiedBetween=nothing, returnPrimal::Bool=false, augmentedInit=false, world=nothing, parent_target=nothing, method_table=nothing, kwargs...)
 
     tt    = Tuple{map(eltype, types.parameters)...}
     if world === nothing