From 348c3e05be2341cca4c8b2a22a5adeb5ba2d0664 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 7 Apr 2026 08:41:14 +0000 Subject: [PATCH] Rename default_policy to policy for hstu_attention forward --- ...ault_policy.hpp => hstu_attention_fwd_pipeline_policy.hpp} | 2 +- .../hstu_attention_no_softmax_fwd_pipeline.hpp | 4 ++-- .../hstu_attention_no_softmax_fwd_trload_pipeline.hpp | 4 ++-- .../hstu_attention_with_softmax_fwd_pipeline.hpp | 4 ++-- .../hstu_attention_with_softmax_fwd_trload_pipeline.hpp | 4 ++-- 5 files changed, 9 insertions(+), 9 deletions(-) rename example/ck_tile/18_hstu_attention/{hstu_attention_fwd_pipeline_default_policy.hpp => hstu_attention_fwd_pipeline_policy.hpp} (99%) 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_policy.hpp similarity index 99% rename from example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp rename to example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_policy.hpp index 13da5774b7..4fb2feb6e1 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_policy.hpp @@ -17,7 +17,7 @@ namespace ck_tile { -struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy +struct HstuAttentionFwdPipelineQRKSVSPolicy { template CK_TILE_DEVICE static constexpr auto GetNumKVLdsBuffers() diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_pipeline.hpp index 14cdae8103..fd984a207d 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_pipeline.hpp @@ -6,13 +6,13 @@ #include "ck_tile/core.hpp" #include "ck_tile/ops/fmha/block/block_dropout.hpp" -#include "hstu_attention_fwd_pipeline_default_policy.hpp" +#include "hstu_attention_fwd_pipeline_policy.hpp" namespace ck_tile { template + typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy> struct HstuAttentionNoSoftmaxFwdPipelineQRKSVS { using Problem = remove_cvref_t; diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_trload_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_trload_pipeline.hpp index a385443af8..8d2bac5579 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_trload_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_no_softmax_fwd_trload_pipeline.hpp @@ -6,13 +6,13 @@ #include "ck_tile/core.hpp" #include "ck_tile/ops/fmha/block/block_dropout.hpp" -#include "hstu_attention_fwd_pipeline_default_policy.hpp" +#include "hstu_attention_fwd_pipeline_policy.hpp" namespace ck_tile { template + typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy> struct HstuAttentionNoSoftmaxFwdPipelineQRKSVSTrLoad { using Problem = remove_cvref_t; diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp index 049d08ea41..e598a7439c 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp @@ -6,13 +6,13 @@ #include "ck_tile/core.hpp" #include "ck_tile/ops/fmha/block/block_dropout.hpp" -#include "hstu_attention_fwd_pipeline_default_policy.hpp" +#include "hstu_attention_fwd_pipeline_policy.hpp" namespace ck_tile { template + typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy> struct HstuAttentionWithSoftmaxFwdPipelineQRKSVS { using Problem = remove_cvref_t; diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp index 2306f28b5a..b5fdf03d47 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp @@ -6,13 +6,13 @@ #include "ck_tile/core.hpp" #include "ck_tile/ops/fmha/block/block_dropout.hpp" -#include "hstu_attention_fwd_pipeline_default_policy.hpp" +#include "hstu_attention_fwd_pipeline_policy.hpp" namespace ck_tile { template + typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy> struct HstuAttentionWithSoftmaxFwdPipelineQRKSVSTrLoad { using Problem = remove_cvref_t;