ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-03 13:48:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

const blockq

Browse Source
This commit is contained in:
Tianxing Wu
2025-10-23 11:57:36 +00:00
parent 3bb29bfd6c
commit ebf1c4c305
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp
View File

@@ -395,7 +395,8 @@ struct UnifiedAttentionPipeline
void* smem_ptr) const
{
using namespace ck_tile;
constexpr index_t BLOCK_Q = BLOCK_M / num_queries_per_kv;
// TODO do we make the num_queries_per_kv and num_head conexpr???
const index_t BLOCK_Q = BLOCK_M / num_queries_per_kv;
static_assert(
std::is_same_v<QDataType, remove_cvref_t<typename QDramBlockWindowTmp::DataType>> &&