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

Rename the hacked block_gemm_areg_bsmem_creg_v2

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-05-15 10:38:15 +00:00
parent 58e45ec53a
commit 473fbc374b
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack.hpp → example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack_0.hpp
View File

@@ -12,7 +12,7 @@ namespace ck_tile {
// B is block window on shared memory
// C is block distributed tensor
template <typename Problem_, typename Policy_ = BlockGemmARegBSmemCRegV2DefaultPolicy>
struct BlockGemmARegBSmemCRegV2Hack
struct BlockGemmARegBSmemCRegV2Hack_0
{
using Problem = remove_cvref_t<Problem_>;
using Policy = remove_cvref_t<Policy_>;

4
example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp
View File

@@ -6,7 +6,7 @@
#include "ck_tile/core.hpp"
#include "ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qx_ks_vs_custom_policy.hpp"
#include "block_gemm_areg_bsmem_creg_v2_hack.hpp"
#include "block_gemm_areg_bsmem_creg_v2_hack_0.hpp"
namespace ck_tile {
@@ -413,7 +413,7 @@ struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy
decltype(warp_gemm)>;
if constexpr(1 < Problem::kNumGemm0Warps)
return BlockGemmARegBSmemCRegV2Hack<GemmProblem, BlockGemmPolicy>{};
return BlockGemmARegBSmemCRegV2Hack_0<GemmProblem, BlockGemmPolicy>{};
else
return BlockGemmARegBSmemCRegOneWarpV1<GemmProblem, BlockGemmPolicy>{};
}