From 1f5c6f2db1ffe2d55515b73478238bf0596ab1de Mon Sep 17 00:00:00 2001 From: Haocong WANG Date: Wed, 12 Mar 2025 00:15:26 +0800 Subject: [PATCH] reduce test size to avoid timeout on specific silicon (#1966) [ROCm/composable_kernel commit: ba209b9dabd7edf801be3e81d3e9d566a2123bfa] --- .../test_batched_gemm_softmax_gemm_permute_bf16_xdl.cpp | 5 ----- 1 file changed, 5 deletions(-) diff --git a/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_bf16_xdl.cpp b/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_bf16_xdl.cpp index 8e0baede11..8136257a24 100644 --- a/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_bf16_xdl.cpp +++ b/test/batched_gemm_softmax_gemm_permute/test_batched_gemm_softmax_gemm_permute_bf16_xdl.cpp @@ -102,7 +102,6 @@ TYPED_TEST(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, Bench_BF16_Irregul {256, 64, 160, 64, 1, 16}, {1024, 1024, 80, 80, 1, 16}, {1024, 64, 80, 64, 1, 16}, - {4096, 4096, 40, 40, 1, 16}, {4096, 64, 40, 64, 1, 16}}; this->bench_ = true; this->verify_ = false; @@ -118,10 +117,6 @@ TYPED_TEST(TestBatchedGemmMaskingScaleSoftmaxGemmPermuteBF16, Bench_BF16) {512, 512, 128, 128, 48, 16}, {1024, 1024, 64, 64, 48, 16}, {1024, 1024, 128, 128, 48, 16}, - {2048, 2048, 64, 64, 48, 16}, - {2048, 2048, 128, 128, 48, 16}, - {4096, 4096, 64, 64, 48, 16}, - {4096, 4096, 128, 128, 48, 16}, }; this->bench_ = true; this->verify_ = false;