ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename default_policy to policy for hstu_attention forward

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-04-07 08:41:14 +00:00
parent 423cc72bc4
commit 348c3e05be
5 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_default_policy.hpp → example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline_policy.hpp
View File

@@ -17,7 +17,7 @@
namespace ck_tile {
struct HstuAttentionFwdPipelineQRKSVSDefaultPolicy
struct HstuAttentionFwdPipelineQRKSVSPolicy
{
template <typename Problem>
CK_TILE_DEVICE static constexpr auto GetNumKVLdsBuffers()

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

@@ -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 Problem_,
typename Traits_,
typename Policy_ = HstuAttentionFwdPipelineQRKSVSDefaultPolicy>
typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy>
struct HstuAttentionNoSoftmaxFwdPipelineQRKSVS
{
using Problem = remove_cvref_t<Problem_>;

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

@@ -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 Problem_,
typename Traits_,
typename Policy_ = HstuAttentionFwdPipelineQRKSVSDefaultPolicy>
typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy>
struct HstuAttentionNoSoftmaxFwdPipelineQRKSVSTrLoad
{
using Problem = remove_cvref_t<Problem_>;

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

@@ -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 Problem_,
typename Traits_,
typename Policy_ = HstuAttentionFwdPipelineQRKSVSDefaultPolicy>
typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy>
struct HstuAttentionWithSoftmaxFwdPipelineQRKSVS
{
using Problem = remove_cvref_t<Problem_>;

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

@@ -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 Problem_,
typename Traits_,
typename Policy_ = HstuAttentionFwdPipelineQRKSVSDefaultPolicy>
typename Policy_ = HstuAttentionFwdPipelineQRKSVSPolicy>
struct HstuAttentionWithSoftmaxFwdPipelineQRKSVSTrLoad
{
using Problem = remove_cvref_t<Problem_>;