add module to RuntimeGeneratedFunctions call (#2259)

Co-authored-by: Yingbo Ma <[email protected]>
SciML · Sep 20, 2023 · 68dfae6 · 68dfae6
1 parent 475ade8
commit 68dfae6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/systems/pde/pdesystem.jl b/src/systems/pde/pdesystem.jl
@@ -94,6 +94,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
         analytic = nothing,
         analytic_func = nothing,
         gui_metadata = nothing,
+        eval_module = @__MODULE__,
         checks::Union{Bool, Int} = true,
         name)
         if checks == true || (checks & CheckUnits) > 0
@@ -114,7 +115,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
                     p = ps isa SciMLBase.NullParameters ? [] : map(a -> a.first, ps)
                     args = vcat(DestructuredArgs(p), args)
                     ex = Func(args, [], eq.rhs) |> toexpr
-                    eq.lhs => drop_expr(@RuntimeGeneratedFunction(ex))
+                    eq.lhs => drop_expr(@RuntimeGeneratedFunction(eval_module, ex))
                 end
             end
         end