github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix format (#574)

pull/673/head
wky 2022-03-31 15:47:51 +08:00 committed by binmakeswell
parent dfe423ae42
commit 174b9c1d85
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h
View File

@ -10,9 +10,8 @@
using namespace std; using namespace std;
template <typename T> template <typename T> class Normalize_Layer {
class Normalize_Layer { public:
public:
struct Config { struct Config {
uint32_t hidden_dim; uint32_t hidden_dim;
bool use_mean; bool use_mean;
@ -58,7 +57,7 @@ class Normalize_Layer {
inline bool use_mean() const { return config_.use_mean; } inline bool use_mean() const { return config_.use_mean; }
private: private:
Config config_; Config config_;
T *vars_; T *vars_;
T *means_; T *means_;