Skip to content

GitLab

Commit d6f5ef88 authored by doubleHU's avatar doubleHU Committed by Frank Lee
Browse files

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

parent 6d6c01e9
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 6 deletions
colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu
View file @ d6f5ef88
...@@ -135,9 +135,10 @@ __global__ void bias_add_transform_20314(T *output, const T *input, ...@@ -135,9 +135,10 @@ __global__ void bias_add_transform_20314(T *output, const T *input,
const T *bias, int dim_3, int dim_4); const T *bias, int dim_3, int dim_4);
template <> 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 id0 = blockIdx.x;
int id1 = blockIdx.y; int id1 = blockIdx.y;
int id2 = blockIdx.z; int id2 = blockIdx.z;
...@@ -173,9 +174,10 @@ bias_add_transform_20314<float>(float *output, const float *input, ...@@ -173,9 +174,10 @@ bias_add_transform_20314<float>(float *output, const float *input,
} }
template <> 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 id0 = blockIdx.x;
int id1 = blockIdx.y; int id1 = blockIdx.y;
int id2 = blockIdx.z; int id2 = blockIdx.z;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment