[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code stype (#629)

01cc941e · puck_WCR · binmakeswell · c1bed0d9 · 01cc941e
Commit 01cc941e authored Apr 01, 2022 by puck_WCR Committed by binmakeswell Apr 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 8 deletions

colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu ...alai/kernel/cuda_native/csrc/kernels/transform_kernels.cu +6 -8

No files found.
--- a/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu
+++ b/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu
@@ -135,10 +135,9 @@ __global__ void bias_add_transform_20314(T *output, const T *input,
                                         const T *bias, int dim_3, int dim_4);

 template <>
-__global__ void bias_add_transform_20314<float>(float *output,
-                                                const float *input,
-                                                const float *bias, int dim_3,
-                                                int dim_4) {
+__global__ void
+bias_add_transform_20314<float>(float *output, const float *input,
+                                const float *bias, int dim_3, int dim_4) {
  int id0 = blockIdx.x;
  int id1 = blockIdx.y;
  int id2 = blockIdx.z;
@@ -174,10 +173,9 @@ __global__ void bias_add_transform_20314<float>(float *output,
 }

 template <>
-__global__ void bias_add_transform_20314<__half>(__half *output,
-                                                 const __half *input,
-                                                 const __half *bias, int dim_3,
-                                                 int dim_4) {
+__global__ void
+bias_add_transform_20314<__half>(__half *output, const __half *input,
+                                 const __half *bias, int dim_3, int dim_4) {
  int id0 = blockIdx.x;
  int id1 = blockIdx.y;
  int id2 = blockIdx.z;