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

Add cuda 11.2 driver and runtime inc files

Browse Source
This commit is contained in:
Nathan Luehr 2021-01-11 17:28:39 -06:00
parent 21ee78c7c9
commit ee68122aa1
4 changed files with 4974 additions and 2 deletions

2816
tensorflow/stream_executor/cuda/cuda_11_2.inc Normal file
View File

File diff suppressed because it is too large Load Diff

2152
tensorflow/stream_executor/cuda/cuda_runtime_11_2.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -101,6 +101,8 @@ typedef void(CUDA_CB* CUhostFn)(void* userData);
#include "tensorflow/stream_executor/cuda/cuda_10_1.inc"
#elif CUDA_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cuda_10_2.inc"
#else
#elif CUDA_VERSION < 11020
#include "tensorflow/stream_executor/cuda/cuda_11_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cuda_11_2.inc"
#endif

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

@ -59,8 +59,10 @@ cudaError_t GetSymbolNotFoundError() {
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_1.inc"
#elif CUDART_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_2.inc"
#else
#elif CUDART_VERSION < 11020
#include "tensorflow/stream_executor/cuda/cuda_runtime_11_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cuda_runtime_11_2.inc"
#endif
#undef __dv
#undef __CUDA_DEPRECATED