"driver/include/device_tensor.hpp" did not exist on "31ded4ac4bc524acdbf897ffff094d7e7cbed991"
Commit 7804a412 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 633199a1
...@@ -47,8 +47,8 @@ void logsoftmax(hipStream_t stream, const argument& result, const argument& arg, ...@@ -47,8 +47,8 @@ void logsoftmax(hipStream_t stream, const argument& result, const argument& arg,
auto data_idx = batch_idx; auto data_idx = batch_idx;
// load data to lds and compute the batch max // load data to lds and compute the batch max
size_t remaining_item_num = batch_item_num; size_t remaining_item_num = batch_item_num;
size_t round_item_num = (batch_item_num + block_size - 1) / block_size * block_size; size_t round_item_num = (batch_item_num + block_size - 1) / block_size * block_size;
lds_data[block_size] = input_ptr[0]; lds_data[block_size] = input_ptr[0];
for(size_t i = thr_idx; i < round_item_num; i += block_size) for(size_t i = thr_idx; i < round_item_num; i += block_size)
{ {
if(i < batch_item_num) if(i < batch_item_num)
......
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