Unverified Commit a7e63bfa authored by Max Podkorytov's avatar Max Podkorytov
Browse files

roll back splitkv

parent f9627ef4
...@@ -94,8 +94,6 @@ struct BlockFmhaFwdSplitKVCombinePipeline ...@@ -94,8 +94,6 @@ struct BlockFmhaFwdSplitKVCombinePipeline
{ {
constexpr std::array occupancy{2, 2, 2, 2, 2, 1}; constexpr std::array occupancy{2, 2, 2, 2, 2, 1};
return occupancy[detail::log2<kMaxSplits>::value - 2]; return occupancy[detail::log2<kMaxSplits>::value - 2];
} else if constexpr(kHeadDimV <= 512) {
return 1;
} }
} }
}(); }();
......
...@@ -96,10 +96,6 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVS ...@@ -96,10 +96,6 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVS
{ {
return 1; return 1;
} }
else if constexpr(kQKHeaddim <= 512)
{
return 1;
}
} }
}(); }();
......
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