experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

address comments on commit fc58d58923

Browse Source
This commit is contained in:
Bas Aarts 2020-05-08 07:58:05 -07:00
parent fc58d58923
commit cd2f4d1628
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/stream_executor/cuda/cudart_stub.cc
View File

@ -132,9 +132,9 @@ extern __host__ __device__ unsigned CUDARTAPI __cudaPushCallConfiguration(
} }
extern char CUDARTAPI __cudaInitModule(void **fatCubinHandle) { extern char CUDARTAPI __cudaInitModule(void **fatCubinHandle) {
using FuncPtr = cudaError_t(CUDARTAPI *)(void **fatCubinHandle); using FuncPtr = char(CUDARTAPI *)(void **fatCubinHandle);
static auto func_ptr = LoadSymbol<FuncPtr>("__cudaInitModule"); static auto func_ptr = LoadSymbol<FuncPtr>("__cudaInitModule");
if (!func_ptr) return GetSymbolNotFoundError(); if (!func_ptr) return 0;
return func_ptr(fatCubinHandle); return func_ptr(fatCubinHandle);
} }