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

Fix gemm unit tests: adding conditional tuple selection, based on pipeline type, on diverging input params

Browse Source
This commit is contained in:
Damien Lejeune
2025-07-31 07:17:06 +00:00
parent 8e878f3914
commit 370ce9435b
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -82,14 +82,20 @@ class TestCkTileGemmPipeline : public ::testing::Test
using CDataType = std::tuple_element_t<6, Tuple>;
static constexpr auto Scheduler = std::tuple_element_t<7, Tuple>::value;
static constexpr auto PipelineType = std::tuple_element_t<8, Tuple>::value;
static constexpr bool SkipALds = std::tuple_element_t<9, Tuple>::value;
static constexpr bool SkipBLds = std::tuple_element_t<10, Tuple>::value;
static constexpr bool Persistent = (PipelineType == GemmPipelineType::CompV3)
? std::tuple_element_t<9, Tuple>::value
: std::false_type::value;
static constexpr bool SkipALds = (PipelineType == GemmPipelineType::Mem)
? std::tuple_element_t<9, Tuple>::value
: std::false_type::value;
static constexpr bool SkipBLds = (PipelineType == GemmPipelineType::Mem)
? std::tuple_element_t<10, Tuple>::value
: std::false_type::value;
using DsLayout = ck_tile::tuple<>;
using DsDataType = ck_tile::tuple<>;
static constexpr bool Persistent =
ck_tile::tuple_element_or_default_t<Tuple, 9, std::false_type>::value;
// TODO: expose tile size through test t-param ?
template <bool PadM, bool PadN, bool PadK, bool Preshuffle>