diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py index 2b262eb09d21ded0e0f9c80c2e765f17e5f5b5b1..82309d945546956b18602f7ad06277b491f5da91 100644 --- a/arraycontext/impl/pytato/__init__.py +++ b/arraycontext/impl/pytato/__init__.py @@ -103,7 +103,8 @@ class PytatoPyOpenCLArrayContext(ArrayContext): def call_loopy(self, program, **kwargs): import pyopencl.array as cla from pytato.loopy import call_loopy - entrypoint, = set(program.callables_table) + + entrypoint = program.default_entrypoint.name # thaw frozen arrays kwargs = {kw: (self.thaw(arg) if isinstance(arg, cla.Array) else arg)