From a0b85a1bf228fce923e2c8772439e99bfe147c42 Mon Sep 17 00:00:00 2001 From: Vidyasagar Ananthan Date: Fri, 5 Sep 2025 14:57:59 -0700 Subject: [PATCH] Fixing tile engine tests after recent refactoring. (#2791) * Fixing tile engine tests after recent refactoring. * Fixing line break error. [ROCm/composable_kernel commit: 60ea94f4fe2d86fa147d30194beb37bb84971fed] --- Jenkinsfile | 38 ++++++-------------------------------- 1 file changed, 6 insertions(+), 32 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 96af656c16..dbbc2d6f64 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -1357,22 +1357,9 @@ pipeline { -D GEMM_MULTI_D_DATATYPE="fp16" \ -D GEMM_MULTI_D_LAYOUT="rcrr;rrrr;crrr;ccrr" \ -DCMAKE_CXX_FLAGS=" -O3 " .. && \ - ninja -j64 benchmark_gemm_fp8_rcr && \ - ./bin/benchmark_gemm_fp8_rcr && \ - ninja -j64 benchmark_gemm_fp16_rcr && \ - ./bin/benchmark_gemm_fp16_rcr && \ - ninja -j64 benchmark_gemm_fp8_crr && \ - ./bin/benchmark_gemm_fp8_crr && \ - ninja -j64 benchmark_gemm_fp16_crr && \ - ./bin/benchmark_gemm_fp16_crr && \ - ninja -j64 benchmark_gemm_fp8_ccr && \ - ./bin/benchmark_gemm_fp8_ccr && \ - ninja -j64 benchmark_gemm_fp16_ccr && \ - ./bin/benchmark_gemm_fp16_ccr && \ - ninja -j64 benchmark_gemm_fp8_rrr && \ - ./bin/benchmark_gemm_fp8_rrr && \ - ninja -j64 benchmark_gemm_fp16_rrr && \ - ./bin/benchmark_gemm_fp16_rrr && \ + ninja -j64 benchmark_gemm_all && \ + python3 ../tile_engine/ops/gemm/gemm_benchmark.py bin --problem-sizes "1024,1024,1024" \ + --warmup 5 --repeat 5 --verbose --json results.json && \ ninja -j64 benchmark_gemm_multi_d_fp16_rrrr && \ ./bin/benchmark_gemm_multi_d_fp16_rrrr && \ ninja -j64 benchmark_gemm_multi_d_fp16_ccrr && \ @@ -1405,22 +1392,9 @@ pipeline { -D GEMM_MULTI_D_DATATYPE="fp16" \ -D GEMM_MULTI_D_LAYOUT="rcrr;rrrr;crrr;ccrr" \ -DCMAKE_CXX_FLAGS=" -O3 " .. && \ - ninja -j64 benchmark_gemm_fp8_rcr && \ - ./bin/benchmark_gemm_fp8_rcr && \ - ninja -j64 benchmark_gemm_fp16_rcr && \ - ./bin/benchmark_gemm_fp16_rcr && \ - ninja -j64 benchmark_gemm_fp8_crr && \ - ./bin/benchmark_gemm_fp8_crr && \ - ninja -j64 benchmark_gemm_fp16_crr && \ - ./bin/benchmark_gemm_fp16_crr && \ - ninja -j64 benchmark_gemm_fp8_ccr && \ - ./bin/benchmark_gemm_fp8_ccr && \ - ninja -j64 benchmark_gemm_fp16_ccr && \ - ./bin/benchmark_gemm_fp16_ccr && \ - ninja -j64 benchmark_gemm_fp8_rrr && \ - ./bin/benchmark_gemm_fp8_rrr && \ - ninja -j64 benchmark_gemm_fp16_rrr && \ - ./bin/benchmark_gemm_fp16_rrr && \ + ninja -j64 benchmark_gemm_all && \ + python3 ../tile_engine/ops/gemm/gemm_benchmark.py bin --problem-sizes "1024,1024,1024" \ + --warmup 5 --repeat 5 --verbose --json results.json && \ ninja -j64 benchmark_gemm_multi_d_fp16_rrrr && \ ./bin/benchmark_gemm_multi_d_fp16_rrrr && \ ninja -j64 benchmark_gemm_multi_d_fp16_ccrr && \