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

2022-05-15 09:00:09 +08:00 · 2022-05-15 09:00:09 +08:00 · 1dc1b6fa00
parent 0e922da874
commit 1dc1b6fa00
1 changed files with 4 additions and 3 deletions
--- a/colossalai/kernel/cuda_native/csrc/kernels/include/cross_entropy_layer.h
+++ b/colossalai/kernel/cuda_native/csrc/kernels/include/cross_entropy_layer.h
@ -8,7 +8,8 @@

 #include "cuda_util.h"

-template <typename T> class CrossEntropyLayer {
+template <typename T>
+class CrossEntropyLayer {
 public:
  CrossEntropyLayer(float epsilon, int padding_idx, int max_batch_tokens);