Skip to content
Snippets Groups Projects
Unverified Commit 8e5dbc5f authored by Matthias Diener's avatar Matthias Diener Committed by GitHub
Browse files

call_loopy: correct entrypoint (#58)

parent be0f569e
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment