Debug print in wave_ops

iree-org · Feb 14, 2025 · 2b97ac1 · 2b97ac1
1 parent 8465d7c
commit 2b97ac1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/iree/turbine/kernel/ops/wave_ops.py b/iree/turbine/kernel/ops/wave_ops.py
@@ -1233,7 +1233,9 @@ def align_index(self, constraints: list["Constraint"]) -> None:
         from ..wave.utils import align_index_vars, is_shared_mem_access
 
         if is_shared_mem_access(self):
+            print(f"MMA before {self.index} with constraints {constraints}")
             self.index = align_index_vars(self.index, constraints)
+            print(f"MMA  after {self.index}")
 
     def transform_index_backwards(
         self, index: dict[IndexSymbol, IndexSequence], arg: fx.Node