diff --git a/test/ck_tile/gemm/test_gemm_pipeline_kernel_types.hpp b/test/ck_tile/gemm/test_gemm_pipeline_kernel_types.hpp index ed6e074206..09df777ec0 100644 --- a/test/ck_tile/gemm/test_gemm_pipeline_kernel_types.hpp +++ b/test/ck_tile/gemm/test_gemm_pipeline_kernel_types.hpp @@ -86,9 +86,4 @@ using KernelTypesPersistent = ::testing::Types< std::tuple< Row, Col, Row, F16, F16, F32, F16, Intrawave, CompV3, NonPersistent> >; -using KernelTypesPersistent = ::testing::Types< - std::tuple< Row, Col, Row, F16, F16, F32, F16, Intrawave, CompV3, Persistent>, - std::tuple< Row, Col, Row, F16, F16, F32, F16, Intrawave, CompV3, NonPersistent> ->; - // clang-format on diff --git a/test/ck_tile/gemm/test_gemm_pipeline_util.hpp b/test/ck_tile/gemm/test_gemm_pipeline_util.hpp index d3b1d276ae..4de18647f9 100644 --- a/test/ck_tile/gemm/test_gemm_pipeline_util.hpp +++ b/test/ck_tile/gemm/test_gemm_pipeline_util.hpp @@ -83,15 +83,17 @@ 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; - 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; + static constexpr bool Persistent = std::conditional_t, + std::false_type>::value; + + static constexpr bool SkipALds = std::conditional_t, + std::false_type>::value; + + static constexpr bool SkipBLds = std::conditional_t, + std::false_type>::value; using DsLayout = ck_tile::tuple<>; using DsDataType = ck_tile::tuple<>; @@ -121,9 +123,6 @@ class TestCkTileGemmPipeline : public ::testing::Test constexpr bool DoubleSmemBuffer = (PipelineType == GemmPipelineType::CompV4) ? true : false; - constexpr bool kSkipALds = SkipALds; - constexpr bool kSkipBLds = SkipBLds; - // TODO: For now - but this should also be a test parameter constexpr bool TransposeC = false; @@ -156,8 +155,8 @@ class TestCkTileGemmPipeline : public ::testing::Test Persistent, NumWaveGroup, preshuffle, - kSkipALds, - kSkipBLds>; + SkipALds, + SkipBLds>; using GemmPipelineProblem = ck_tile::GemmPipelineProblem;