diff --git a/build_tools/ci/cpu_comparison/run.py b/build_tools/ci/cpu_comparison/run.py index a8979f5cb..d34f6af95 100755 --- a/build_tools/ci/cpu_comparison/run.py +++ b/build_tools/ci/cpu_comparison/run.py @@ -1796,6 +1796,26 @@ def __init__(self): additional_labels=["I8UKernel"], ) ) + self.register( + Matmul( + 64, + 64, + 64, + "bf16", + "f32", + name_suffix="4rows_8cols_npu4_pack_peel_4_level_tiling_ukernel", + use_ukernel=True, + tile_pipeline="pack-peel-4-level-tiling", + run_on_target=["npu4"], + aie_compilation_flags=[ + "--iree-amdaie-num-rows=4", + "--iree-amdaie-num-cols=8", + "--iree-amd-aie-additional-peano-opt-flags=-O3", + "--iree-amdaie-enable-function-outlining=True", + ], + use_chess=True, + ) + ) # Matmul test on 2(rows)x2(cols) cores self.register(