From 01cc941e1d7ef66f2c8c200d3ca677dd589bed97 Mon Sep 17 00:00:00 2001
From: puck_WCR <46049915+WANG-CR@users.noreply.github.com>
Date: Fri, 1 Apr 2022 20:20:54 +0800
Subject: [PATCH] [NFC] polish
 colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code stype
 (#629)

---
 .../cuda_native/csrc/kernels/transform_kernels.cu  | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu b/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu
index d389d57e1..d03084b22 100644
--- 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;