"composable_kernel/include/utility/utility.hpp" did not exist on "a0584426ff5b6b8b448c971b97c9b1a4d86ba010"
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 ...@@ -45,7 +45,8 @@ void add(hipStream_t stream, const argument& result, const argument& arg1, const
auto last_dim = sr.lens().back() / 2; auto last_dim = sr.lens().back() / 2;
int block_size = 1024; int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size; 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 else
{ {
......
...@@ -45,7 +45,8 @@ void mul(hipStream_t stream, const argument& result, const argument& arg1, const ...@@ -45,7 +45,8 @@ void mul(hipStream_t stream, const argument& result, const argument& arg1, const
auto last_dim = sr.lens().back() / 2; auto last_dim = sr.lens().back() / 2;
int block_size = 1024; int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size; 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 else
{ {
......
...@@ -78,14 +78,16 @@ void mul_add(hipStream_t stream, ...@@ -78,14 +78,16 @@ void mul_add(hipStream_t stream,
auto n_dim = lens.size(); auto n_dim = lens.size();
int block_size = 1024; int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size; 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 else
{ {
int factor = lens[1]; 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 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