Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
c5206c90
Commit
c5206c90
authored
Feb 03, 2025
by
Qianfeng Zhang
Browse files
Remove using KLoadOnce in qx_ks_vs_custom_policy
parent
87b206fb
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
3 deletions
+1
-3
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qx_ks_vs_custom_policy.hpp
...a/pipeline/block_fmha_pipeline_qx_ks_vs_custom_policy.hpp
+1
-3
No files found.
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qx_ks_vs_custom_policy.hpp
View file @
c5206c90
...
@@ -287,14 +287,12 @@ struct BlockFmhaPipelineQXKSVSCustomPolicy : BlockFmhaPipelineQXCustomPolicy<QLo
...
@@ -287,14 +287,12 @@ struct BlockFmhaPipelineQXKSVSCustomPolicy : BlockFmhaPipelineQXCustomPolicy<QLo
// preload V-slice for next unroll using single LDS buffer
// preload V-slice for next unroll using single LDS buffer
static
constexpr
bool
AsyncCopy
=
AsyncCopy_
;
static
constexpr
bool
AsyncCopy
=
AsyncCopy_
;
static
constexpr
bool
KLoadOnce
=
AsyncCopy
;
using
QXPolicy
=
BlockFmhaPipelineQXCustomPolicy
<
QLoadOnce_
>
;
using
QXPolicy
=
BlockFmhaPipelineQXCustomPolicy
<
QLoadOnce_
>
;
template
<
typename
Problem
>
template
<
typename
Problem
>
CK_TILE_DEVICE
static
constexpr
auto
GetNumKLdsBuffers
()
CK_TILE_DEVICE
static
constexpr
auto
GetNumKLdsBuffers
()
{
{
if
constexpr
(
KLoadOnce
)
if
constexpr
(
AsyncCopy
)
{
{
return
2
;
return
2
;
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment