From f1e989f5aa468e39520ad0b623b4442a900666d7 Mon Sep 17 00:00:00 2001
From: Xu Kai <xukai16@foxmail.com>
Date: Mon, 16 May 2022 10:48:42 +0800
Subject: [PATCH] [NFC] polish
 colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h code style

---
 .../kernel/cuda_native/csrc/kernels/include/dropout.h     | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h b/colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h
index c2a4f7c20a35..563a7fe284a3 100644
--- a/colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h
+++ b/colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h
@@ -3,12 +3,14 @@
 #include <cuda.h>
 #include <cuda_fp16.h>
 #include <stdio.h>
+
 #include <string>
 
 #include "kernels.h"
 
-template <typename T> class Dropout {
-public:
+template <typename T>
+class Dropout {
+ public:
   struct Config {
     float ratio;
     bool training;
@@ -88,7 +90,7 @@ template <typename T> class Dropout {
 
   void SetTrainingMode(bool training) { _config.training = training; }
 
-private:
+ private:
   uint8_t *_mask;
   Config _config;
 };