From 02e459c35374735662c39da589f590bf12651f15 Mon Sep 17 00:00:00 2001 From: huxin711 Date: Tue, 12 Jul 2022 20:26:28 +0800 Subject: [PATCH] [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code style --- .../cuda_native/csrc/kernels/transform_kernels.cu | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu b/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu index d03084b22e12..d389d57e11b6 100644 --- a/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu +++ b/colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu @@ -135,9 +135,10 @@ __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 *output, const float *input, - const float *bias, int dim_3, int dim_4) { +__global__ void bias_add_transform_20314(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; @@ -173,9 +174,10 @@ bias_add_transform_20314(float *output, const float *input, } 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;