ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-12 02:05:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the compiling error in HSTU_SCHED_BATCH_AS_FIRST_GRID_DIM=0 path

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-06-30 09:10:12 +00:00
parent 80d7b8dce7
commit 4667be2bb0
3 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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
}
}

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

@@ -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
}

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

@@ -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
}
}