Skip to content

Commit 68dfae6

Browse files
xtalaxYingboMa
andauthored
add module to RuntimeGeneratedFunctions call (#2259)
Co-authored-by: Yingbo Ma <[email protected]>
1 parent 475ade8 commit 68dfae6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/systems/pde/pdesystem.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
9494
analytic = nothing,
9595
analytic_func = nothing,
9696
gui_metadata = nothing,
97+
eval_module = @__MODULE__,
9798
checks::Union{Bool, Int} = true,
9899
name)
99100
if checks == true || (checks & CheckUnits) > 0
@@ -114,7 +115,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
114115
p = ps isa SciMLBase.NullParameters ? [] : map(a -> a.first, ps)
115116
args = vcat(DestructuredArgs(p), args)
116117
ex = Func(args, [], eq.rhs) |> toexpr
117-
eq.lhs => drop_expr(@RuntimeGeneratedFunction(ex))
118+
eq.lhs => drop_expr(@RuntimeGeneratedFunction(eval_module, ex))
118119
end
119120
end
120121
end

0 commit comments

Comments
 (0)