mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 02:02:46 +00:00
This reverts commit c5014a86765c598215bf2b58f89c3a7c70f92bac.
[ROCm/composable_kernel commit: 722c22fb15]
This commit is contained in:
@@ -122,9 +122,6 @@ struct BlockFmhaBatchPrefillPipelineQRKSVSAsync
|
||||
{
|
||||
if constexpr(kPadSeqLenK && BiasEnum == BlockAttentionBiasEnum::ELEMENTWISE_BIAS)
|
||||
return 1;
|
||||
// use larger K/V LDS buffer size will lower the occupancy
|
||||
else if constexpr(64 <= kK0 || 64 <= kK1)
|
||||
return 1;
|
||||
else
|
||||
return 2;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user