ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] Fix batched_gemm tests for gfx950 (#2869)

Browse Source 
[ROCm/composable_kernel commit: 427dca076b]
This commit is contained in:
aledudek
2025-09-18 01:43:41 +02:00
committed by GitHub
parent 3f44e675e4
commit ed6fd3d53a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/ck_tile/batched_gemm/test_batched_gemm_ut_cases.inc
View File

@@ -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,