ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-09 08:37:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor conditional test parameters

Browse Source
This commit is contained in:
Damien Lejeune
2025-08-02 11:33:57 +00:00
parent 15bbf0da5e
commit a1af89245a
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
test/ck_tile/gemm/test_gemm_pipeline_util.hpp
View File

@@ -83,18 +83,20 @@ class TestCkTileGemmPipeline : public ::testing::Test
static constexpr auto Scheduler = std::tuple_element_t<7, Tuple>::value;
static constexpr auto PipelineType = std::tuple_element_t<8, Tuple>::value;
template <GemmPipelineType TargetPipeline, ck_tile::index_t I, typename T>
static constexpr bool get_param_or_default(bool default_value)
{
if constexpr(PipelineType == TargetPipeline && std::tuple_size_v<T> > I)
{
return std::tuple_element_t<I, T>::value;
}
return default_value;
}
static constexpr bool Persistent =
(PipelineType == GemmPipelineType::CompV3 && std::tuple_size_v<Tuple> > 9)
? std::tuple_element_t<9, Tuple>::value
: std::false_type::value;
static constexpr bool SkipALds =
(PipelineType == GemmPipelineType::Mem && std::tuple_size_v<Tuple> > 9)
? std::tuple_element_t<9, Tuple>::value
: std::false_type::value;
static constexpr bool SkipBLds =
(PipelineType == GemmPipelineType::Mem && std::tuple_size_v<Tuple> > 10)
? std::tuple_element_t<10, Tuple>::value
: std::false_type::value;
get_param_or_default<GemmPipelineType::CompV3, 9, Tuple>(false);
static constexpr bool SkipALds = get_param_or_default<GemmPipelineType::Mem, 9, Tuple>(false);
static constexpr bool SkipBLds = get_param_or_default<GemmPipelineType::Mem, 10, Tuple>(false);
using DsLayout = ck_tile::tuple<>;
using DsDataType = ck_tile::tuple<>;