From 174b9c1d85fae486909ad7e9769dd234a2ee494d Mon Sep 17 00:00:00 2001 From: wky <64853922+wangkuangyi@users.noreply.github.com> Date: Thu, 31 Mar 2022 15:47:51 +0800 Subject: [PATCH] fix format (#574) --- .../cuda_native/csrc/kernels/include/normalize_layer.h | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h b/colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h index 22e16fe90..ded5c0fdc 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h +++ b/colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h @@ -10,9 +10,8 @@ using namespace std; -template -class Normalize_Layer { - public: +template class Normalize_Layer { +public: struct Config { uint32_t hidden_dim; bool use_mean; @@ -58,7 +57,7 @@ class Normalize_Layer { inline bool use_mean() const { return config_.use_mean; } - private: +private: Config config_; T *vars_; T *means_;