diff --git a/test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc b/test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc index b2f965764d..035377734b 100644 --- a/test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc +++ b/test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc @@ -23,13 +23,16 @@ TYPED_TEST(TestCkTileBatchedGemm, Basic) std::vector gemmParams{{256, 256, 256, 1}, {256, 256, 256, 2}, {256, 256, 512, 2}, - {256, 256, 128, 2}, {256, 256, 64, 2}, {256, 256, 64, 3}, {256, 256, 64, 4}, {256, 256, 64, 8}, {256, 256, 64, 16}}; + if(ck_tile::get_device_name() != "gfx950") { + gemmParams.emplace_back(256, 256, 128, 2); + } + for(auto& params : gemmParams) { std::vector strideConfigs{{params.K,