Commit c5206c90 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Remove using KLoadOnce in qx_ks_vs_custom_policy

parent 87b206fb
......@@ -287,14 +287,12 @@ struct BlockFmhaPipelineQXKSVSCustomPolicy : BlockFmhaPipelineQXCustomPolicy<QLo
// preload V-slice for next unroll using single LDS buffer
static constexpr bool AsyncCopy = AsyncCopy_;
static constexpr bool KLoadOnce = AsyncCopy;
using QXPolicy = BlockFmhaPipelineQXCustomPolicy<QLoadOnce_>;
template <typename Problem>
CK_TILE_DEVICE static constexpr auto GetNumKLdsBuffers()
{
if constexpr(KLoadOnce)
if constexpr(AsyncCopy)
{
return 2;
}
......
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