Commit 5459e4d8 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 2d5e45b8
......@@ -45,7 +45,8 @@ void add(hipStream_t stream, const argument& result, const argument& arg1, const
auto last_dim = sr.lens().back() / 2;
int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size;
add_kernel<<<block_num, block_size>>>(arg1.data(), arg2.data(), last_dim, result.data(), elem_num);
add_kernel<<<block_num, block_size>>>(
arg1.data(), arg2.data(), last_dim, result.data(), elem_num);
}
else
{
......
......@@ -45,7 +45,8 @@ void mul(hipStream_t stream, const argument& result, const argument& arg1, const
auto last_dim = sr.lens().back() / 2;
int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size;
mul_kernel<<<block_num, block_size>>>(arg1.data(), arg2.data(), last_dim, result.data(), elem_num);
mul_kernel<<<block_num, block_size>>>(
arg1.data(), arg2.data(), last_dim, result.data(), elem_num);
}
else
{
......
......@@ -78,14 +78,16 @@ void mul_add(hipStream_t stream,
auto n_dim = lens.size();
int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size;
if (n_dim == 2)
if(n_dim == 2)
{
mul_add_kernel_dim3<<<block_num, block_size>>>(arg1.data(), arg2.data(), arg3.data(), last_dim, result.data(), elem_num);
mul_add_kernel_dim3<<<block_num, block_size>>>(
arg1.data(), arg2.data(), arg3.data(), last_dim, result.data(), elem_num);
}
else
{
int factor = lens[1];
mul_add_kernel_dim4<<<block_num, block_size>>>(arg1.data(), arg2.data(), arg3.data(), factor, last_dim, result.data(), elem_num);
mul_add_kernel_dim4<<<block_num, block_size>>>(
arg1.data(), arg2.data(), arg3.data(), factor, last_dim, result.data(), elem_num);
}
}
else
......
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