diff --git a/example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack.hpp b/example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack_0.hpp similarity index 99% rename from example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack.hpp rename to example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack_0.hpp index f0145e7b85..1b895c81a9 100644 --- a/example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack.hpp +++ b/example/ck_tile/18_hstu_attention/block_gemm_areg_bsmem_creg_v2_hack_0.hpp @@ -12,7 +12,7 @@ namespace ck_tile { // B is block window on shared memory // C is block distributed tensor template -struct BlockGemmARegBSmemCRegV2Hack +struct BlockGemmARegBSmemCRegV2Hack_0 { using Problem = remove_cvref_t; using Policy = remove_cvref_t; diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp index 9ea83ec606..bcd8ad4a12 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp @@ -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{}; + return BlockGemmARegBSmemCRegV2Hack_0{}; else return BlockGemmARegBSmemCRegOneWarpV1{}; }