From ebf1c4c305db789197bd90b96b65cb1850993d15 Mon Sep 17 00:00:00 2001 From: Tianxing Wu Date: Thu, 23 Oct 2025 11:57:36 +0000 Subject: [PATCH] const blockq --- .../unified_attention/pipeline/unified_attention_pipeline.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp b/include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp index b8534b7a6a..ec4b3355de 100644 --- a/include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp +++ b/include/ck_tile/ops/unified_attention/pipeline/unified_attention_pipeline.hpp @@ -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> &&