From 2b97ac1c08750bc9c91023af6f92a11688e5ca24 Mon Sep 17 00:00:00 2001 From: Nicolas Vasilache Date: Fri, 14 Feb 2025 11:45:25 -0800 Subject: [PATCH] Debug print in wave_ops --- iree/turbine/kernel/ops/wave_ops.py | 2 ++ 1 file changed, 2 insertions(+) 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