fix format (#574)

174b9c1d · wky · binmakeswell · dfe423ae · 174b9c1d
Commit 174b9c1d authored Mar 31, 2022 by wky Committed by binmakeswell Apr 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

colossalai/kernel/cuda_native/csrc/kernels/include/normalize_layer.h ...kernel/cuda_native/csrc/kernels/include/normalize_layer.h +3 -4

No files found.
--- 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 <typename T>
+template <typename T> class Normalize_Layer {
-class Normalize_Layer {
+public:
- 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_;