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

Merge commit '427dca076b228f7db32c7a0046652d5b37e88aa2' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-18 00:32:29 +00:00
parent 649a5b43f4
commit 5aa0aacc81
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,