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
372d603b
Commit
372d603b
authored
Feb 03, 2025
by
Aviral Goel
Browse files
space added
parent
785003e0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
4 deletions
+6
-4
include/ck_tile/ops/fmha/kernel/fmha_flex_fwd_kernel.hpp
include/ck_tile/ops/fmha/kernel/fmha_flex_fwd_kernel.hpp
+6
-4
No files found.
include/ck_tile/ops/fmha/kernel/fmha_flex_fwd_kernel.hpp
View file @
372d603b
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-202
4
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 2018-202
5
, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
#pragma once
...
@@ -20,7 +20,10 @@
...
@@ -20,7 +20,10 @@
namespace
ck_tile
{
namespace
ck_tile
{
template
<
typename
FmhaPipeline_
,
typename
EpiloguePipeline_
,
typename
ScoreModFunction_
,
typename
PreSoftmaxFunction_
>
template
<
typename
FmhaPipeline_
,
typename
EpiloguePipeline_
,
typename
ScoreModFunction_
,
typename
PreSoftmaxFunction_
>
struct
FmhaFwdKernel
struct
FmhaFwdKernel
{
{
using
FmhaPipeline
=
ck_tile
::
remove_cvref_t
<
FmhaPipeline_
>
;
using
FmhaPipeline
=
ck_tile
::
remove_cvref_t
<
FmhaPipeline_
>
;
...
@@ -1317,8 +1320,7 @@ struct FmhaFwdKernel
...
@@ -1317,8 +1320,7 @@ struct FmhaFwdKernel
};
};
auto
pre_softmax_def
=
PreSoftmaxFunction_
{};
auto
pre_softmax_def
=
PreSoftmaxFunction_
{};
auto
pre_softmax_arg
=
[
pre_softmax_def
](
auto
pre_softmax_arg
=
[
pre_softmax_def
](
typename
PreSoftmaxFunction_
::
TScore
s
)
{
typename
PreSoftmaxFunction_
::
TScore
s
)
{
return
pre_softmax_def
(
s
);
return
pre_softmax_def
(
s
);
};
};
...
...
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