ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

roll back splitkv

Browse Source 
[ROCm/composable_kernel commit: 60113859fa]
This commit is contained in:
Max Podkorytov
2024-12-18 02:00:39 +00:00
parent 63fdc5c84e
commit 62b1f2e830
2 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_combine_pipeline.hpp
View File

@@ -95,8 +95,6 @@ struct BlockFmhaFwdSplitKVCombinePipeline
{
constexpr std::array occupancy{2, 2, 2, 2, 2, 1};
return occupancy[detail::log2<kMaxSplits>::value - 2];
} else if constexpr(kHeadDimV <= 512) {
return 1;
}
}
}();

4
include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs.hpp
View File

@@ -96,10 +96,6 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVS
{
return 1;
}
else if constexpr(kQKHeaddim <= 512)
{
return 1;
}
}
}();