From ed6fd3d53aa57781dd23f1d2e8b260526745493e Mon Sep 17 00:00:00 2001
From: aledudek <aleksander.dudek@amd.com>
Date: Thu, 18 Sep 2025 01:43:41 +0200
Subject: [PATCH] [CK_TILE] Fix batched_gemm tests for gfx950 (#2869)

[ROCm/composable_kernel commit: 427dca076b228f7db32c7a0046652d5b37e88aa2]
---
 test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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> 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<StrideConfig> strideConfigs{{params.K,