From a81656ed219a8fa5c02605496704b638f8c5d705 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Wed, 2 Jun 2021 13:47:53 -0500
Subject: [PATCH] kernel_callables_v3-edit2 for loopy, for_each_kernel
---
arraycontext/impl/pytato.py | 4 ++--
requirements.txt | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index beba602..5291c99 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -341,12 +341,12 @@ class PytatoArrayContext(ArrayContext):
output_spec)
def transform_loopy_program(self, prg):
- from loopy.program import iterate_over_kernels_if_given_program
+ from loopy.translation_unit import for_each_kernel
nwg = 48
nwi = (16, 2)
- @iterate_over_kernels_if_given_program
+ @for_each_kernel
def gridify(knl):
# {{{ Pattern matching inames
diff --git a/requirements.txt b/requirements.txt
index 066db94..b1beba7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ git+https://github.com/inducer/pymbolic.git#egg=pymbolic
git+https://github.com/inducer/pyopencl.git#egg=pyopencl
git+https://github.com/inducer/islpy.git#egg=islpy
-git+https://github.com/inducer/loopy.git#egg=loopy
+git+https://github.com/inducer/loopy@kernel_callables_v3-edit2.git#egg=loopy
git+https://github.com/kaushikcfd/pytato.git@call_loopy#egg=pytato
git+https://github.com/inducer/meshmode.git#egg=meshmode
--
GitLab