Commit 372d603b authored by Aviral Goel's avatar Aviral Goel
Browse files

space added

parent 785003e0
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved. // Copyright (c) 2018-2025, 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);
}; };
......
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