diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 3ad3d70a16f3ac56d207cb73b271877702d58b98..8d3b678f5ff58082eb57a749efa9c1f02e434d33 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -527,8 +527,11 @@ class PytatoPyOpenCLArrayContext(_BasePytatoArrayContext): transformed_dag, function_name) from arraycontext.loopy import _DEFAULT_LOOPY_OPTIONS + opts = _DEFAULT_LOOPY_OPTIONS + assert opts.return_dict + pt_prg = pt.generate_loopy(transformed_dag, - options=_DEFAULT_LOOPY_OPTIONS, + options=opts, cl_device=self.queue.device, function_name=function_name, target=self.get_target()) diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py index 9e7a72c40d2a91d2d3aaa29e65d0797a6447c656..c49968792461a1a38f0002d90fe233573306a494 100644 --- a/arraycontext/impl/pytato/compile.py +++ b/arraycontext/impl/pytato/compile.py @@ -414,9 +414,12 @@ class LazilyPyOpenCLCompilingFunctionCaller(BaseLazilyCompilingFunctionCaller): with ProcessLogger(logger, f"generate_loopy for '{prg_id}'"): from arraycontext.loopy import _DEFAULT_LOOPY_OPTIONS + opts = _DEFAULT_LOOPY_OPTIONS + assert opts.return_dict + pytato_program = pt.generate_loopy( pt_dict_of_named_arrays, - options=_DEFAULT_LOOPY_OPTIONS, + options=opts, function_name=_prg_id_to_kernel_name(prg_id), target=self.actx.get_target(), )