kernel_callables_v3-edit2 for loopy, for_each_kernel

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
 

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