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

632e94ab · Xu Kai · binmakeswell · 22d1df22 · 632e94ab
Commit 632e94ab authored May 16, 2022 by Xu Kai Committed by binmakeswell May 17, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h +5 -3

No files found.
--- 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 @@ public:

  void SetTrainingMode(bool training) { _config.training = training; }

-private:
+ private:
  uint8_t *_mask;
  Config _config;
 };