Commit 93eae2df authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 9fc184e5
...@@ -16,11 +16,10 @@ inline __device__ void reduce_max(T* data_ptr, size_t block_size, size_t thr_idx ...@@ -16,11 +16,10 @@ inline __device__ void reduce_max(T* data_ptr, size_t block_size, size_t thr_idx
while(true) while(true)
{ {
auto stride = (item_num + 1) / 2; auto stride = (item_num + 1) / 2;
auto size = item_num / 2; auto size = item_num / 2;
for (size_t i = thr_idx; i < size; i += block_size) for(size_t i = thr_idx; i < size; i += block_size)
{ {
data_ptr[i] = data_ptr[i] = ::max(to_hip_type(data_ptr[i]), to_hip_type(data_ptr[i + stride]));
::max(to_hip_type(data_ptr[i]), to_hip_type(data_ptr[i + stride]));
} }
__syncthreads(); __syncthreads();
item_num = stride; item_num = stride;
...@@ -44,8 +43,8 @@ inline __device__ void reduce_sum(T* data_ptr, size_t block_size, size_t thr_idx ...@@ -44,8 +43,8 @@ inline __device__ void reduce_sum(T* data_ptr, size_t block_size, size_t thr_idx
while(true) while(true)
{ {
auto stride = (item_num + 1) / 2; auto stride = (item_num + 1) / 2;
auto size = item_num / 2; auto size = item_num / 2;
for (size_t i = thr_idx; i < size; i += block_size) for(size_t i = thr_idx; i < size; i += block_size)
{ {
data_ptr[i] += data_ptr[i + stride]; data_ptr[i] += data_ptr[i + stride];
} }
......
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