From 49d16600da8feb91aa5dc7bc0fc3459407f704a9 Mon Sep 17 00:00:00 2001 From: Abhishek Varma Date: Wed, 15 Jan 2025 13:42:25 +0000 Subject: [PATCH] [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 --- build_tools/ci/cpu_comparison/run.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/build_tools/ci/cpu_comparison/run.py b/build_tools/ci/cpu_comparison/run.py index ef5546745..aada185e3 100755 --- 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"