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
1f15fbea
Commit
1f15fbea
authored
Jan 26, 2025
by
Qianfeng Zhang
Browse files
Re-format
parent
21dc4596
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
.../ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
+4
-4
No files found.
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
View file @
1f15fbea
...
@@ -326,6 +326,8 @@ struct BlockFmhaPipelineQRKSVSAsync
...
@@ -326,6 +326,8 @@ struct BlockFmhaPipelineQRKSVSAsync
clear_tile
(
s_acc
);
// initialize C
clear_tile
(
s_acc
);
// initialize C
static_for
<
0
,
k0_loops
-
1
,
1
>
{}([
&
](
auto
i_k0
)
{
static_for
<
0
,
k0_loops
-
1
,
1
>
{}([
&
](
auto
i_k0
)
{
__builtin_amdgcn_sched_barrier
(
0
);
if
constexpr
(
i_k0
>
0
&&
i_k0
<
k0_loops
-
1
)
if
constexpr
(
i_k0
>
0
&&
i_k0
<
k0_loops
-
1
)
{
{
if
constexpr
(
i_k0
%
2
==
1
)
if
constexpr
(
i_k0
%
2
==
1
)
...
@@ -464,10 +466,8 @@ struct BlockFmhaPipelineQRKSVSAsync
...
@@ -464,10 +466,8 @@ struct BlockFmhaPipelineQRKSVSAsync
block_tile_reduce_sync
(
m_local
,
f_max
,
bool_constant
<
false
>
{});
block_tile_reduce_sync
(
m_local
,
f_max
,
bool_constant
<
false
>
{});
const
auto
m_old
=
m
;
// m{j-1}
const
auto
m_old
=
m
;
// m{j-1}
tile_elementwise_inout
([](
auto
&
e0
,
auto
e1
,
auto
e2
)
{
e0
=
max
(
e1
,
e2
);
},
tile_elementwise_inout
(
m
,
[](
auto
&
e0
,
auto
e1
,
auto
e2
)
{
e0
=
max
(
e1
,
e2
);
},
m
,
m_old
,
m_local
);
// m{j}
m_old
,
m_local
);
// m{j}
auto
p_compute
=
make_static_distributed_tensor
<
SMPLComputeDataType
>
(
auto
p_compute
=
make_static_distributed_tensor
<
SMPLComputeDataType
>
(
s
.
get_tile_distribution
());
// Pcompute{j}
s
.
get_tile_distribution
());
// Pcompute{j}
...
...
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