diff --git a/iree/turbine/kernel/ops/wave_ops.py b/iree/turbine/kernel/ops/wave_ops.py
index f6959d1f2..e8cd21e63 100644
--- 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