Restore loop inference fallback

README.rst

@@ -24,7 +24,7 @@ implementations for:
 - Profiling
 
 ``arraycontext`` started life as an array abstraction for use with the
-`meshmode <https://documen.tician.de/meshmode/>`__ unstrucuted discretization
+`meshmode <https://documen.tician.de/meshmode/>`__ unstructured discretization
 package.
 
 Distributed under the MIT license.

arraycontext/impl/pyopencl/__init__.py

@@ -325,9 +325,31 @@ def transform_loopy_program(self, t_unit):
 
         all_inames = default_entrypoint.all_inames()
 
+        # FIXME: This could be much smarter.
         inner_iname = None
 
-        if "i0" in all_inames:
+        from meshmode.transform_metadata import FirstAxisIsElementsTag
+
+        if (len(default_entrypoint.instructions) == 1
+                and isinstance(default_entrypoint.instructions[0], lp.Assignment)
+                and any(isinstance(tag, FirstAxisIsElementsTag)
+                    # FIXME: Firedrake branch lacks kernel tags
+                    for tag in getattr(default_entrypoint, "tags", ()))):
+            stmt, = default_entrypoint.instructions
+
+            out_inames = [v.name for v in stmt.assignee.index_tuple]
+            assert out_inames
+            outer_iname = out_inames[0]
+            if len(out_inames) >= 2:
+                inner_iname = out_inames[1]
+
+        elif "iel" in all_inames:
+            outer_iname = "iel"
+
+            if "idof" in all_inames:
+                inner_iname = "idof"
+
+        elif "i0" in all_inames:
             outer_iname = "i0"
 
             if "i1" in all_inames:

requirements.txt

@@ -7,3 +7,5 @@ git+https://github.com/inducer/islpy.git#egg=islpy
 
 git+https://github.com/inducer/loopy.git#egg=loopy
 git+https://github.com/inducer/pytato.git#egg=pytato
+
+git+https://github.com/inducer/meshmode.git#egg=meshmode