Commit ba29c1f2 authored by Fenglei Tian's avatar Fenglei Tian

code style

parent 555d5929
...@@ -36,13 +36,13 @@ namespace ngraph ...@@ -36,13 +36,13 @@ namespace ngraph
{ {
std::string name = "abs"; std::string name = "abs";
// Create an instance of nvrtcProgram with the code string. // Create an instance of nvrtcProgram with the code string.
if (CudaFunctionPool::Instance().get(name) == nullptr) if (CudaFunctionPool::instance().get(name) == nullptr)
{ {
const char* opts[] = {"--gpu-architecture=compute_35", const char* opts[] = {"--gpu-architecture=compute_35",
"--relocatable-device-code=true"}; "--relocatable-device-code=true"};
std::string kernel; std::string kernel;
CudaKernelBuilder::get_1_element_op(name, "float", "fabsf", kernel); CudaKernelBuilder::get_1_element_op(name, "float", "fabsf", kernel);
CudaFunctionPool::Instance().set( CudaFunctionPool::instance().set(
name, CudaFunctionBuilder::get("cuda_" + name, kernel, 2, opts)); name, CudaFunctionBuilder::get("cuda_" + name, kernel, 2, opts));
} }
...@@ -53,7 +53,7 @@ namespace ngraph ...@@ -53,7 +53,7 @@ namespace ngraph
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
......
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