diff options
Diffstat (limited to 'tensorflow/core/kernels/cuda_solvers.cc')
-rw-r--r-- | tensorflow/core/kernels/cuda_solvers.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cuda_solvers.cc b/tensorflow/core/kernels/cuda_solvers.cc index a857bd3ce4..a59baaa96f 100644 --- a/tensorflow/core/kernels/cuda_solvers.cc +++ b/tensorflow/core/kernels/cuda_solvers.cc @@ -151,7 +151,7 @@ CudaSolver::CudaSolver(OpKernelContext* context) : context_(context) { reinterpret_cast<const cudaStream_t*>(context->op_device_context() ->stream() ->implementation() - ->CudaStreamMemberHack())); + ->GpuStreamMemberHack())); cuda_stream_ = *cu_stream_ptr; HandleMap* handle_map = CHECK_NOTNULL(GetHandleMapSingleton()); auto it = handle_map->find(cuda_stream_); |