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

[CK] Add command option instance_index and param_mask to run partial ck test (#2889)

Browse Source 
* [CK] Add command option instance_index and param_mask to run partial ck test

Many CK test are instance test. it will loop all instance in the instance library. It causes test often out-of-time if we run test on simulator/emulator.
This PR add option instance_index and param_mask to reduce the workload of instance test

instance_index: only run test 1 available instance with specified index.
param_mask: filter the embedded parameter with specified mask

* fix CI error

* fix clang format

---------

Co-authored-by: illsilin_amdeng <Illia.Silin@amd.com>
This commit is contained in:
linqunAMD
2025-09-30 23:24:40 +08:00
committed by GitHub
parent 28ad8ae5d8
commit e78a897ec0
113 changed files with 2804 additions and 704 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/gemm_split_k/test_gemm_splitk_util.hpp
View File

@@ -15,6 +15,8 @@
#include "include/ck/utility/data_type.hpp"
#include "profiler/profile_gemm_splitk_impl.hpp"
extern ck::index_t param_mask;
extern ck::index_t instance_index;
namespace ck {
namespace test {
@@ -48,8 +50,13 @@ class TestGemmSplitK : public testing::Test
const int StrideB,
const int StrideC)
{
for(auto kb : k_batches_)
for(size_t i = 0; i < k_batches_.size(); i++)
{
if((param_mask & (1 << i)) == 0)
{
continue;
}
auto kb = k_batches_[i];
RunSingle(M, N, K, StrideA, StrideB, StrideC, kb);
}
}
@@ -82,7 +89,8 @@ class TestGemmSplitK : public testing::Test
StrideC,
kbatch,
n_warmup,
n_iter);
n_iter,
instance_index);
EXPECT_TRUE(pass);
}
};