Commit 21d1070c authored by fenglei.tian's avatar fenglei.tian

clang format

parent ba29c1f2
...@@ -52,18 +52,17 @@ namespace ngraph ...@@ -52,18 +52,17 @@ namespace ngraph
d_ptr_out = (CUdeviceptr)out; d_ptr_out = (CUdeviceptr)out;
void* args_list[] = {&d_ptr_in, &d_ptr_out, &count}; void* args_list[] = {&d_ptr_in, &d_ptr_out, &count};
CUDA_SAFE_CALL( CUDA_SAFE_CALL(cuLaunchKernel(*CudaFunctionPool::instance().get(name).get(),
cuLaunchKernel(*CudaFunctionPool::instance().get(name).get(), count,
count, 1,
1, 1, // grid dim
1, // grid dim 1,
1, 1,
1, 1, // block dim
1, // block dim 0,
0, NULL, // shared mem and stream
NULL, // shared mem and stream args_list,
args_list, 0)); // arguments
0)); // arguments
CUDA_SAFE_CALL(cuCtxSynchronize()); // Retrieve and print output. CUDA_SAFE_CALL(cuCtxSynchronize()); // Retrieve and print output.
} }
} }
......
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