[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style (#964)

7106a399 · Yuer867 · binmakeswell · 5bd80b7d · 7106a399
Commit 7106a399 authored May 15, 2022 by Yuer867 Committed by binmakeswell May 17, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

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

No files found.
--- a/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
+++ b/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
@@ -10,8 +10,9 @@

 using namespace std;

-template <typename T> class Softmax {
-public:
+template <typename T>
+class Softmax {
+ public:
  struct Config {
    size_t nhead;
    Config(size_t nhead) : nhead(nhead) {}
@@ -36,6 +37,6 @@ public:

  void reset_size(size_t nhead) { config_.nhead = nhead; }

-private:
+ private:
  Config config_;
 };