diff --git a/impl/normalization/normalize/normalize_common_impl.h b/impl/normalization/normalize/normalize_common_impl.h index 737da6c9821d4b0358d0eabf90c761c9db737e43..0d4177e0465754063d489d2e0dde8f95cb7e3425 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 322c9de971d71ed684f7e6be85a77793300f61c5..86e1710e124b4db50915e9ac794a3daa86f31ce8 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