From cd2f4d16282d36c47547ba72d762ad967ce1d024 Mon Sep 17 00:00:00 2001 From: Bas Aarts Date: Fri, 8 May 2020 07:58:05 -0700 Subject: [PATCH] address comments on commit fc58d58923534e461d735a9a8b460d2dc8691ae5 --- tensorflow/stream_executor/cuda/cudart_stub.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/stream_executor/cuda/cudart_stub.cc b/tensorflow/stream_executor/cuda/cudart_stub.cc index 1d7a4e378ba..3b9e0f2937b 100644 --- a/tensorflow/stream_executor/cuda/cudart_stub.cc +++ b/tensorflow/stream_executor/cuda/cudart_stub.cc @@ -132,9 +132,9 @@ extern __host__ __device__ unsigned CUDARTAPI __cudaPushCallConfiguration( } extern char CUDARTAPI __cudaInitModule(void **fatCubinHandle) { - using FuncPtr = cudaError_t(CUDARTAPI *)(void **fatCubinHandle); + using FuncPtr = char(CUDARTAPI *)(void **fatCubinHandle); static auto func_ptr = LoadSymbol("__cudaInitModule"); - if (!func_ptr) return GetSymbolNotFoundError(); + if (!func_ptr) return 0; return func_ptr(fatCubinHandle); }