diff --git a/examples/dagrt-fusion.py b/examples/dagrt-fusion.py index e188438fb4dc54ee41a59607b32ea64a2852009e..27e35abd82161e528481344eb28a6f0f81c22839 100755 --- a/examples/dagrt-fusion.py +++ b/examples/dagrt-fusion.py @@ -52,7 +52,7 @@ from pymbolic.mapper.evaluator import EvaluationMapper \ from pytools import memoize from grudge import sym, bind, DGDiscretizationWithBoundaries -from leap.rk import LSRK4Method +from leap.rk import LSRK4MethodBuilder from pyopencl.tools import ( # noqa pytest_generate_tests_for_pyopencl as pytest_generate_tests) @@ -267,7 +267,7 @@ class RK4TimeStepperBase(object): def set_up_stepper(self, discr, field_var_name, sym_rhs, num_fields, function_registry=base_function_registry, exec_mapper_factory=ExecutionMapper): - dt_method = LSRK4Method(component_id=field_var_name) + dt_method = LSRK4MethodBuilder(component_id=field_var_name) dt_code = dt_method.generate() self.field_var_name = field_var_name self.state_name = f"input_{field_var_name}" diff --git a/grudge/shortcuts.py b/grudge/shortcuts.py index bb12689bf65208c3a48bb299f2a44c6651c5b58b..475f72d24be204627b114813493ce623c431f9c4 100644 --- a/grudge/shortcuts.py +++ b/grudge/shortcuts.py @@ -29,10 +29,10 @@ import pyopencl as cl def set_up_rk4(field_var_name, dt, fields, rhs, t_start=0): - from leap.rk import LSRK4Method + from leap.rk import LSRK4MethodBuilder from dagrt.codegen import PythonCodeGenerator - dt_method = LSRK4Method(component_id=field_var_name) + dt_method = LSRK4MethodBuilder(component_id=field_var_name) dt_code = dt_method.generate() dt_stepper_class = PythonCodeGenerator("TimeStep").get_class(dt_code) dt_stepper = dt_stepper_class({""+dt_method.component_id: rhs})