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

Merge pull request #39281 from bas-aarts:bas_fix_tf_missing_symbol_dbg_build

Browse Source 
PiperOrigin-RevId: 311400520
Change-Id: Id5f6292f57e68871f92673727dbf8ed82a17d147
This commit is contained in:
TensorFlower Gardener 2020-05-13 14:10:53 -07:00
commit ffb7db1a81
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -131,6 +131,13 @@ extern __host__ __device__ unsigned CUDARTAPI __cudaPushCallConfiguration(
return func_ptr(gridDim, blockDim, sharedMem, stream);
}
extern char CUDARTAPI __cudaInitModule(void **fatCubinHandle) {
using FuncPtr = char(CUDARTAPI *)(void **fatCubinHandle);
static auto func_ptr = LoadSymbol<FuncPtr>("__cudaInitModule");
if (!func_ptr) return 0;
return func_ptr(fatCubinHandle);
}
#if CUDART_VERSION >= 10010
extern void CUDARTAPI __cudaRegisterFatBinaryEnd(void **fatCubinHandle) {
using FuncPtr = void(CUDARTAPI *)(void **fatCubinHandle);