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

clang format

parent 65386ce5
...@@ -160,7 +160,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in ...@@ -160,7 +160,7 @@ void softmax(hipStream_t stream, const argument& result, const argument& arg, in
auto half2_block_size = compute_block_size(batch_item_num, 1024); auto half2_block_size = compute_block_size(batch_item_num, 1024);
int block_num = batch_shape.elements(); int block_num = batch_shape.elements();
int shared_size = batch_item_num * 2 * result.get_shape().type_size(); int shared_size = batch_item_num * 2 * result.get_shape().type_size();
half2_block_size = half2_block_size / 4; half2_block_size = half2_block_size / 4;
softmax_kernel<<<block_num, half2_block_size, shared_size, stream>>>( softmax_kernel<<<block_num, half2_block_size, shared_size, stream>>>(
arg.data(), batch_item_num, half2_block_size, result.data()); arg.data(), batch_item_num, half2_block_size, result.data());
} }
......
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