Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add ukernel to e2e CI
Browse files Browse the repository at this point in the history
Abhishek-Varma committed Jan 27, 2025
1 parent cfd9773 commit def3a83
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions build_tools/ci/cpu_comparison/run.py
Original file line number Diff line number Diff line change
@@ -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(

0 comments on commit def3a83

Please sign in to comment.