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

d6f5ef88 · doubleHU · Frank Lee · 6d6c01e9 · d6f5ef88
Commit d6f5ef88 authored Jul 13, 2022 by doubleHU Committed by Frank Lee Jul 13, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 6 deletions

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

No files found.
--- 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
+__global__ void bias_add_transform_20314<float>(float *output,
-bias_add_transform_20314<float>(float *output, const float *input,
+                                                const float *input,
-                                const float *bias, int dim_3, int dim_4) {
+                                                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>(float *output, const float *input,
 }
 template <>
-__global__ void
+__global__ void bias_add_transform_20314<__half>(__half *output,
-bias_add_transform_20314<__half>(__half *output, const __half *input,
+                                                 const __half *input,
-                                 const __half *bias, int dim_3, int dim_4) {
+                                                 const __half *bias, int dim_3,
+                                                 int dim_4) {
  int id0 = blockIdx.x;
  int id1 = blockIdx.y;
  int id2 = blockIdx.z;