From 29a0670744ccb1e915ef3255df12ef867fcb3c9a Mon Sep 17 00:00:00 2001 From: rocking Date: Mon, 8 Apr 2024 10:03:48 +0000 Subject: [PATCH] Remove remove_cvref_t --- .../ops/fmha/kernel/fmha_fwd_kernel.hpp | 22 +++++++------------ 1 file changed, 8 insertions(+), 14 deletions(-) diff --git a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp index 7e3cfae9db..9f54008457 100644 --- a/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp +++ b/include/ck_tile/ops/fmha/kernel/fmha_fwd_kernel.hpp @@ -38,20 +38,14 @@ struct FmhaFwdKernel using VLayout = ck_tile::remove_cvref_t; using ElementFunctions = typename FmhaPipeline::Problem::ElementFunctions; - // using QElementFunction = ck_tile::remove_cvref_t; using KElementFunction = - // ck_tile::remove_cvref_t; using VElementFunction - // = ck_tile::remove_cvref_t; using - // BiasElementFunction = - // ck_tile::remove_cvref_t; - // using LSEElementFunction = - // ck_tile::remove_cvref_t; - // using SAccElementFunction = - // ck_tile::remove_cvref_t; - using PComputeElementFunction = - ck_tile::remove_cvref_t; - using OAccElementFunction = - ck_tile::remove_cvref_t; + // using QElementFunction = ElementFunctions::QElementFunction; + // using KElementFunction = ElementFunctions::KElementFunction; + // using VElementFunction = ElementFunctions::VElementFunction; + // using BiasElementFunction = ElementFunctions::BiasElementFunction; + // using LSEElementFunction = ElementFunctions::LSEElementFunction; + // using SAccElementFunction = ElementFunctions::SAccElementFunction; + using PComputeElementFunction = ElementFunctions::PComputeElementFunction; + using OAccElementFunction = ElementFunctions::OAccElementFunction; static constexpr bool kIsGroupMode = FmhaPipeline::kIsGroupMode; static constexpr bool kPadSeqLenQ = FmhaPipeline::kPadSeqLenQ;