From 4667be2bb01b7f13c66ce622043ff3b42b32ea34 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 30 Jun 2026 09:10:12 +0000 Subject: [PATCH] Fix the compiling error in HSTU_SCHED_BATCH_AS_FIRST_GRID_DIM=0 path --- .../ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp | 4 +--- .../hstu_attention_fwd_splitkv_combine_kernel.hpp | 4 +--- .../18_hstu_attention/hstu_attention_fwd_splitkv_kernel.hpp | 4 +--- 3 files changed, 3 insertions(+), 9 deletions(-) diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp index cf9d6bab52..dac1e3a888 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_kernel.hpp @@ -608,9 +608,7 @@ struct HstuAttentionFwdKernel #if HSTU_SCHED_BATCH_AS_FIRST_GRID_DIM return dim3(batch_size_, nhead_, num_tile_in_seqlen); #else - return dim3(num_tile_in_seqlen), - nhead_, - batch_size_); + return dim3(num_tile_in_seqlen, nhead_, batch_size_); #endif } } diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_combine_kernel.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_combine_kernel.hpp index cce0ae8d87..c93102f1f0 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_combine_kernel.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_combine_kernel.hpp @@ -232,9 +232,7 @@ struct HstuAttentionFwdSplitKVCombineKernel #if HSTU_SCHED_BATCH_AS_FIRST_GRID_DIM return dim3(batch_size_, nhead_, num_tile_in_seqlen); #else - return dim3(num_tile_in_seqlen), - nhead_, - batch_size_); + return dim3(num_tile_in_seqlen, nhead_, batch_size_); #endif } diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_kernel.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_kernel.hpp index f1a03a6a8d..add49574de 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_kernel.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_splitkv_kernel.hpp @@ -569,9 +569,7 @@ struct HstuAttentionFwdSplitKVKernel #if HSTU_SCHED_BATCH_AS_FIRST_GRID_DIM return dim3(batch_size_, nhead_, num_tile_in_seqlen * num_splits); #else - return dim3(num_tile_in_seqlen*num_splits), - nhead_, - batch_size_); + return dim3(num_tile_in_seqlen * num_splits, nhead_, batch_size_); #endif } }