[CI] Add e2e test for 4x8 AIE Array on Strix

-- Adds e2e Matmul 1024x1204x1024 i32 test for 4x8 AIE array on Strix. Signed-off-by: Abhishek Varma <abhvarma@amd.com>
nod-ai · Jan 15, 2025 · 49d1660 · 49d1660
1 parent e29f65b
commit 49d1660
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/build_tools/ci/cpu_comparison/run.py b/build_tools/ci/cpu_comparison/run.py
@@ -1665,6 +1665,22 @@ def __init__(self):
                 use_chess=True,
             )
         )
+        self.register(
+            Matmul(
+                1024,
+                1024,
+                1024,
+                "i32",
+                "i32",
+                name_suffix="chess_npu4",
+                run_on_target=["npu4"],
+                aie_compilation_flags=[
+                    "--iree-amdaie-num-rows=4",
+                    "--iree-amdaie-num-cols=8",
+                ],
+                use_chess=True,
+            )
+        )
 
         for target in ["npu1_4col", "npu4"]:
             use_chess = target == "npu4"