Skip to content

Commit a700c63

Browse files
committed
debug print
1 parent d2a4e35 commit a700c63

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

.github/workflows/ci.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ jobs:
8585
curl -L -O https://tiker.net/ci-support-v0
8686
. ./ci-support-v0
8787
build_py_project_in_conda_env
88+
export PYTEST_FLAGS=-sv
8889
test_py_project
8990
9091
examples3:

arraycontext/impl/pytato/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1014,7 +1014,9 @@ def transform_loopy_program(
10141014
#. Once the kernel boundaries are inferred, global temporaries are aliased
10151015
to reduce the memory peak memory used by the transformed program.
10161016
"""
1017-
return self._parallelize_across_device(t_unit)
1017+
t_unit = self._parallelize_across_device(t_unit)
1018+
print(t_unit, flush=True)
1019+
return t_unit
10181020

10191021

10201022
# {{{ PytatoJAXArrayContext

0 commit comments

Comments
 (0)