From 8e5dbc5fef1c355e63b9712ecf201b545c67c294 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Tue, 6 Jul 2021 13:53:07 -0500
Subject: [PATCH] call_loopy: correct entrypoint (#58)
---
arraycontext/impl/pytato/__init__.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/arraycontext/impl/pytato/__init__.py b/arraycontext/impl/pytato/__init__.py
index 2b262eb..82309d9 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)
--
GitLab