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

clang format

parent 7da8748c
......@@ -67,7 +67,6 @@ __global__ void add_gelu_kernel(void* a, void* b, int n_dim, void* r, int n)
}
}
void add_gelu(hipStream_t stream,
const argument& result,
const argument& arg1,
......@@ -84,7 +83,8 @@ void add_gelu(hipStream_t stream,
auto last_dim = sr.lens().back() / 2;
int block_size = 1024;
int block_num = (elem_num + block_size - 1) / block_size;
add_gelu_kernel<<<block_num, block_size>>>(arg1.data(), arg2.data(), last_dim, result.data(), elem_num);
add_gelu_kernel<<<block_num, block_size>>>(
arg1.data(), arg2.data(), 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