fix format (#567)

9420d3ae · YuliangLiu0306 · binmakeswell · 0f1da44e · 9420d3ae
Commit 9420d3ae authored Mar 31, 2022 by YuliangLiu0306 Committed by binmakeswell Apr 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 7 deletions

colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h +4 -7

No files found.
--- a/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
+++ b/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
@@ -10,9 +10,8 @@
 using namespace std;
-template <typename T>
+template <typename T> class Softmax {
-class Softmax {
+public:
- public:
  struct Config {
    size_t nhead;
    Config(size_t nhead) : nhead(nhead) {}
@@ -35,10 +34,8 @@ class Softmax {
                              stream);
  }
-  void reset_size(size_t nhead) {
+  void reset_size(size_t nhead) { config_.nhead = nhead; }
-    config_.nhead = nhead;
-  }
- private:
+private:
  Config config_;
 };