Commit d13dcab5 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent ce649bb5
...@@ -77,8 +77,7 @@ inline __device__ void block_reduce_arg(T* data_ptr, ...@@ -77,8 +77,7 @@ inline __device__ void block_reduce_arg(T* data_ptr,
__syncthreads(); __syncthreads();
} }
template <class Op>
template<class Op>
void arg_op(Op op, hipStream_t stream, const argument& result, const argument& arg, int axis) void arg_op(Op op, hipStream_t stream, const argument& result, const argument& arg, int axis)
{ {
auto arg_shape = arg.get_shape(); auto arg_shape = arg.get_shape();
...@@ -124,13 +123,8 @@ void arg_op(Op op, hipStream_t stream, const argument& result, const argument& a ...@@ -124,13 +123,8 @@ void arg_op(Op op, hipStream_t stream, const argument& result, const argument& a
__syncthreads(); __syncthreads();
auto item_num = (remaining_item_num > block_size) ? block_size : remaining_item_num; auto item_num = (remaining_item_num > block_size) ? block_size : remaining_item_num;
block_reduce_arg<type, Op>(lds_data, block_reduce_arg<type, Op>(
lds_index, lds_data, lds_index, op, block_size, thr_idx, item_num, max_block_size);
op,
block_size,
thr_idx,
item_num,
max_block_size);
remaining_item_num -= block_size; remaining_item_num -= block_size;
} }
...@@ -149,4 +143,3 @@ void arg_op(Op op, hipStream_t stream, const argument& result, const argument& a ...@@ -149,4 +143,3 @@ void arg_op(Op op, hipStream_t stream, const argument& result, const argument& a
} // namespace migraphx } // namespace migraphx
#endif #endif
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