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

Add cuda runtime stub for 10.1.

Browse Source 
Following symbols are introduced in 10.1
cudaThreadExchangeStreamCaptureMode
cudaStreamGetCaptureInfo
cudaGraphExecKernelNodeSetParams
__cudaUnregisterFatBinaryEnd

PiperOrigin-RevId: 246544215
This commit is contained in:
Yifei Feng 2019-05-03 11:30:30 -07:00 committed by TensorFlower Gardener
parent 1ff493ed1a
commit 7a466cf7c7
2 changed files with 1855 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1843
tensorflow/stream_executor/cuda/cuda_runtime_10_1.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -49,8 +49,10 @@ cudaError_t GetSymbolNotFoundError() {
// A bunch of new symbols were introduced in version 10
#if CUDA_VERSION <= 9020
#include "tensorflow/stream_executor/cuda/cuda_runtime_9_0.inc"
#else
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_1.inc"
#endif
#undef __dv
#undef __CUDA_DEPRECATED
@ -118,4 +120,13 @@ extern __host__ __device__ unsigned CUDARTAPI __cudaPushCallConfiguration(
if (!func_ptr) return 0;
return func_ptr(gridDim, blockDim, sharedMem, stream);
}
#if CUDA_VERSION >= 10010
extern void CUDARTAPI __cudaUnregisterFatBinaryEnd(void **fatCubinHandle) {
using FuncPtr = void(CUDARTAPI *)(void **fatCubinHandle);
static auto func_ptr = LoadSymbol<FuncPtr>("__cudaUnregisterFatBinaryEnd");
if (!func_ptr) return;
func_ptr(fatCubinHandle);
}
#endif
} // extern "C"