From 37250a3bf0e9adef593b27e23cf8c27a05f3e4a4 Mon Sep 17 00:00:00 2001 From: PengC Date: Thu, 29 May 2025 17:45:41 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9NormalizeTmpTensor=E7=A4=BE?= =?UTF-8?q?=E5=8C=BA=E7=89=88=E6=9C=AC=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- impl/normalization/normalize/normalize_common_impl.h | 9 --------- lib/normalization/normalize_utils.h | 9 +++++++++ 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/impl/normalization/normalize/normalize_common_impl.h b/impl/normalization/normalize/normalize_common_impl.h index 737da6c9..0d4177e0 100644 --- a/impl/normalization/normalize/normalize_common_impl.h +++ b/impl/normalization/normalize/normalize_common_impl.h @@ -22,15 +22,6 @@ namespace AscendC { const float DEFAULT_EPSILON = 1e-5; -template -struct NormalizeTmpTensor { - __aicore__ NormalizeTmpTensor(){}; - LocalTensor tempTensorA; - LocalTensor tempTensorB; - LocalTensor gammaTmpTensor; - LocalTensor betaTmpTensor; -}; - template __aicore__ inline constexpr bool IsDtypeValid() { diff --git a/lib/normalization/normalize_utils.h b/lib/normalization/normalize_utils.h index 322c9de9..86e1710e 100644 --- a/lib/normalization/normalize_utils.h +++ b/lib/normalization/normalize_utils.h @@ -53,5 +53,14 @@ struct NormalizePara { uint32_t rLengthWithPadding; }; +template +struct NormalizeTmpTensor { + __aicore__ NormalizeTmpTensor(){}; + LocalTensor tempTensorA; + LocalTensor tempTensorB; + LocalTensor gammaTmpTensor; + LocalTensor betaTmpTensor; +}; + }; // namespace AscendC #endif // LIB_NORMALIZATION_NORMALIZE_UTILS_H \ No newline at end of file -- Gitee