diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py index 2c1026c9109313f637f096d8d42af2fc6edbde78..5fad96cb14997574c466d82471120447f59378e5 100644 --- a/arraycontext/impl/pytato/compile.py +++ b/arraycontext/impl/pytato/compile.py @@ -225,8 +225,11 @@ class LazilyCompilingFunctionCaller: rec_keyed_map_array_container(_as_dict_of_named_arrays, outputs) + import loopy as lp pytato_program = pt.generate_loopy(dict_of_named_arrays, - options={"return_dict": True}, + options=lp.Options( + return_dict=True, + no_numpy=True), cl_device=self.actx.queue.device) assert isinstance(pytato_program, BoundPyOpenCLProgram)