From 7106a399fc610a2924e09182e581aec5f94fca20 Mon Sep 17 00:00:00 2001
From: Yuer867 <62204893+Yuer867@users.noreply.github.com>
Date: Sun, 15 May 2022 18:06:08 +0800
Subject: [PATCH] [NFC] polish
 colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style
 (#964)

---
 .../kernel/cuda_native/csrc/kernels/include/softmax.h      | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h b/colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h
index 005a36ba1..ec447ad84 100644
--- 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_;
 };