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> &&