ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move static constexpr to structs

Browse Source
This commit is contained in:
Enrico Degregori
2026-01-30 18:08:51 +00:00
parent 0655316e04
commit 17e8f2fa41
3 changed files with 17 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck_tile/core/utility/type_traits.hpp
View File

@@ -109,6 +109,8 @@ using is_known_at_compile_time = is_static<T>;
{ \
};
DEFINE_STATIC_MEMBER_CHECKER(has_bcastpolicy, BCastPolicy);
// FIXME: do we need this anymore?
template <
typename PY,

8
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_base.hpp
View File

@@ -21,7 +21,13 @@ struct GemmPipelineAgBgCrImplBase
using ALayout = remove_cvref_t<std::tuple_element_t<number<0>{}, AsLayout>>;
using BInDataType = remove_cvref_t<std::tuple_element_t<number<0>{}, BsDataType>>;
static constexpr bool IsBCastPolicyBeforeLDSWrite = IsBCastPolicyBeforeLDSWrite_v<Problem>;
static constexpr bool IsBCastPolicyBeforeLDSWrite = [] {
if constexpr(has_bcastpolicy<Problem>::value)
return Problem::BCastPolicy == CastPolicy::BeforeLDSWrite;
else
return false;
}();
using BDataType = std::conditional_t<IsBCastPolicyBeforeLDSWrite, ADataType, BInDataType>;
using BLayout = remove_cvref_t<std::tuple_element_t<number<0>{}, BsLayout>>;

18
include/ck_tile/ops/gemm/pipeline/gemm_universal_pipeline_ag_bg_cr_policy.hpp
View File

@@ -11,16 +11,6 @@
namespace ck_tile {
DEFINE_STATIC_MEMBER_CHECKER(has_bcastpolicy, BCastPolicy);
template <typename Problem>
static constexpr bool IsBCastPolicyBeforeLDSWrite_v = [] {
if constexpr(has_bcastpolicy<Problem>::value)
return Problem::BCastPolicy == CastPolicy::BeforeLDSWrite;
else
return false;
}();
template <typename T, typename = void>
struct has_a_tile_access_pattern : std::false_type
{
@@ -90,6 +80,14 @@ struct UniversalGemmBasePolicy
static constexpr bool is_b_load_tr = false;
#endif
template <typename Problem>
static constexpr bool IsBCastPolicyBeforeLDSWrite_v = [] {
if constexpr(has_bcastpolicy<Problem>::value)
return Problem::BCastPolicy == CastPolicy::BeforeLDSWrite;
else
return false;
}();
static constexpr auto I0 = number<0>{};
static constexpr auto I1 = number<1>{};
static constexpr auto I2 = number<2>{};