Skip to content

GitLab

Commit 346ba760 authored by Po Yen Chen's avatar Po Yen Chen
Browse files

Do not force kHasUnevenSplits=true in group mode

parent af07d650
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_problem.hpp
View file @ 346ba760
...@@ -102,7 +102,7 @@ struct BlockFmhaFwdSplitKVPipelineProblem ...@@ -102,7 +102,7 @@ struct BlockFmhaFwdSplitKVPipelineProblem
static constexpr bool kStoreLSE = Traits::kStoreLSE; static constexpr bool kStoreLSE = Traits::kStoreLSE;
static constexpr bool kDoFp8StaticQuant = Traits::kDoFp8StaticQuant; static constexpr bool kDoFp8StaticQuant = Traits::kDoFp8StaticQuant;
static constexpr bool kIsPagedKV = Traits::kIsPagedKV; static constexpr bool kIsPagedKV = Traits::kIsPagedKV;
static constexpr bool kHasUnevenSplits = kIsGroupMode || Traits::kHasUnevenSplits; static constexpr bool kHasUnevenSplits = Traits::kHasUnevenSplits;
static constexpr index_t kBlockPerCu = Traits::kBlockPerCu; static constexpr index_t kBlockPerCu = Traits::kBlockPerCu;
}; };
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment