ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] Fix Compilation of Flatmm Examples (#3285)

Browse Source
This commit is contained in:
Yi DING
2025-11-26 10:11:43 +08:00
committed by GitHub
parent a54f7b1138
commit c7dce2ac29
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
include/ck_tile/ops/flatmm/pipeline/flatmm_pipeline_agmem_bgmem_creg_v1_policy.hpp Executable file → Normal file
View File

@@ -229,13 +229,13 @@ struct UniversalFlatmmPipelineAgBgCrPolicy
template <typename Problem>
CK_TILE_HOST_DEVICE static constexpr index_t GetSmemPackA()
{
using A = remove_cvref_t<typename Problem::ADataType>;
using BlockGemm = remove_cvref_t<decltype(GetBlockGemm<Problem>())>;
using A = remove_cvref_t<typename Problem::ADataType>;
using BlockFlatmm = remove_cvref_t<decltype(GetBlockFlatmm<Problem>())>;
constexpr index_t KPack = static_cast<index_t>(BlockGemm::Traits::KPack);
constexpr index_t VecElems = static_cast<index_t>(Problem::VectorLoadSize / sizeof(A));
constexpr index_t KPack = BlockFlatmm::BlockPolicy::WarpGemm::kKPerThread;
constexpr index_t VecElems = Problem::VectorLoadSize / sizeof(A);
return (KPack < VecElems) ? KPack : VecElems;
return min(KPack, VecElems);
}
template <typename Problem>