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

space added

parent 785003e0
// 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
......@@ -20,7 +20,10 @@
namespace ck_tile {
template <typename FmhaPipeline_, typename EpiloguePipeline_, typename ScoreModFunction_, typename PreSoftmaxFunction_>
template <typename FmhaPipeline_,
typename EpiloguePipeline_,
typename ScoreModFunction_,
typename PreSoftmaxFunction_>
struct FmhaFwdKernel
{
using FmhaPipeline = ck_tile::remove_cvref_t<FmhaPipeline_>;
......@@ -1317,8 +1320,7 @@ struct FmhaFwdKernel
};
auto pre_softmax_def = PreSoftmaxFunction_{};
auto pre_softmax_arg = [pre_softmax_def](
typename PreSoftmaxFunction_::TScore s) {
auto pre_softmax_arg = [pre_softmax_def](typename PreSoftmaxFunction_::TScore 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