debug print

Author: majosm

.github/workflows/ci.yml

@@ -85,6 +85,8 @@ jobs:
                 curl -L -O https://tiker.net/ci-support-v0
                 . ./ci-support-v0
                 build_py_project_in_conda_env
+                export PYTEST_FLAGS=-sv
+                export CISUPPORT_PYTEST_NRUNNERS=1
                 test_py_project
 
     examples3:

arraycontext/impl/pytato/__init__.py

@@ -1014,7 +1014,12 @@ def transform_loopy_program(
         #. Once the kernel boundaries are inferred, global temporaries are aliased
            to reduce the memory peak memory used by the transformed program.
         """
-        return self._parallelize_across_device(t_unit)
+        t_unit = self._parallelize_across_device(t_unit)
+
+        import sys
+        print(t_unit, file=sys.stderr, flush=True)
+
+        return t_unit
 
 
 # {{{ PytatoJAXArrayContext