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

Add cuda runtime 9.0 API to dlopen wrapper.

Browse Source 
PiperOrigin-RevId: 245304515
This commit is contained in:
Yifei Feng 2019-04-25 14:11:04 -07:00 committed by TensorFlower Gardener
parent 3d04328867
commit 18d3a8c2e8
3 changed files with 1417 additions and 1 deletions
tensorflow/stream_executor/cuda
BUILDcuda_runtime_9_0.inccudart_stub.cc

2
tensorflow/stream_executor/cuda/BUILD
View File

@ -118,7 +118,7 @@ cc_library(
"//tensorflow:using_cuda_clang_with_dynamic_build": ["cudart_stub.cc"],
"//conditions:default": [],
}),
textual_hdrs = ["cuda_runtime_10_0.inc"],
textual_hdrs = glob(["cuda_runtime_*.inc"]),
visibility = ["//visibility:public"],
deps = select({
"//tensorflow:using_cuda_nvcc_with_dynamic_build": [

1410
tensorflow/stream_executor/cuda/cuda_runtime_9_0.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -45,7 +45,13 @@ cudaError_t GetSymbolNotFoundError() {
#define __dv(v)
#define __CUDA_DEPRECATED
// 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
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_0.inc"
#endif
#undef __dv
#undef __CUDA_DEPRECATED