fix merge bug

6482b3d2 · fenglei.tian · f87ed029 · 6482b3d2
Commit 6482b3d2 authored Feb 28, 2018 by fenglei.tian
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

gpu_call_frame.cpp src/ngraph/runtime/gpu/gpu_call_frame.cpp +1 -1

No files found.
--- a/src/ngraph/runtime/gpu/gpu_call_frame.cpp
+++ b/src/ngraph/runtime/gpu/gpu_call_frame.cpp
@@ -47,7 +47,7 @@ runtime::gpu::GPU_CallFrame::GPU_CallFrame(std::shared_ptr<GPU_ExternalFunction>
    }

    // Pass scalars as reference on the Device
-    cublasSetPointerMode(m_cublas_handle, CUBLAS_POINTER_MODE_HOST);
+    cublasSetPointerMode(m_cublas_handle, CUBLAS_POINTER_MODE_DEVICE);
 }

 runtime::gpu::GPU_CallFrame::~GPU_CallFrame()