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

Relax stub include version checking.

Browse Source 
Remove upper bound on version check for latest inc files.
This commit is contained in:
Nathan Luehr 2020-06-11 15:59:21 -05:00
parent f44b07ed4e
commit 3ebc53c394
5 changed files with 20 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/stream_executor/cuda/cublas_stub.cc
View File

@ -63,14 +63,12 @@ typedef enum {} cublasMath_t;
#if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cublas_9_0.inc"
#elif CUDA_VERSION == 10000
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cublas_10_0.inc"
#elif CUDA_VERSION == 10010
#elif CUDA_VERSION < 10020
#include "tensorflow/stream_executor/cuda/cublas_10_1.inc"
#elif CUDA_VERSION == 10020
#elif CUDA_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cublas_10_2.inc"
#elif CUBLAS_VER_MAJOR == 11 && CUBLAS_VER_MINOR == 0
#include "tensorflow/stream_executor/cuda/cublas_11_0.inc"
#else
#error "We have no wrapper for this version."
#include "tensorflow/stream_executor/cuda/cublas_11_0.inc"
#endif

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

@ -95,14 +95,12 @@ typedef void(CUDA_CB* CUhostFn)(void* userData);
#if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cuda_9_0.inc"
#elif CUDA_VERSION == 10000
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cuda_10_0.inc"
#elif CUDA_VERSION <= 10010
#elif CUDA_VERSION < 10020
#include "tensorflow/stream_executor/cuda/cuda_10_1.inc"
#elif CUDA_VERSION <= 10020
#elif CUDA_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cuda_10_2.inc"
#elif CUDA_VERSION <= 11000
#include "tensorflow/stream_executor/cuda/cuda_11_0.inc"
#else
#error "We have no wrapper for this version."
#include "tensorflow/stream_executor/cuda/cuda_11_0.inc"
#endif

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

@ -53,16 +53,14 @@ cudaError_t GetSymbolNotFoundError() {
// A bunch of new symbols were introduced in version 10
#if CUDART_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cuda_runtime_9_0.inc"
#elif CUDART_VERSION == 10000
#elif CUDART_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_0.inc"
#elif CUDART_VERSION == 10010
#elif CUDART_VERSION < 10020
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_1.inc"
#elif CUDART_VERSION == 10020
#elif CUDART_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cuda_runtime_10_2.inc"
#elif CUDART_VERSION == 11000
#include "tensorflow/stream_executor/cuda/cuda_runtime_11_0.inc"
#else
#error "We have no wrapper for this version."
#include "tensorflow/stream_executor/cuda/cuda_runtime_11_0.inc"
#endif
#undef __dv
#undef __CUDA_DEPRECATED

10
tensorflow/stream_executor/cuda/cusolver_stub.cc
View File

@ -53,14 +53,12 @@ cusolverStatus_t GetSymbolNotFoundError() {
#if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cusolver_dense_9_0.inc"
#elif CUDA_VERSION == 10000
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cusolver_dense_10_0.inc"
#elif CUDA_VERSION == 10010
#elif CUDA_VERSION < 10020
#include "tensorflow/stream_executor/cuda/cusolver_dense_10_1.inc"
#elif CUDA_VERSION == 10020
#elif CUDA_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cusolver_dense_10_2.inc"
#elif CUDA_VERSION == 11000
#include "tensorflow/stream_executor/cuda/cusolver_dense_11_0.inc"
#else
#error "We don't have a wrapper for this version."
#include "tensorflow/stream_executor/cuda/cusolver_dense_11_0.inc"
#endif

10
tensorflow/stream_executor/cuda/cusparse_stub.cc
View File

@ -53,14 +53,12 @@ cusparseStatus_t GetSymbolNotFoundError() {
#if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cusparse_9_0.inc"
#elif CUDA_VERSION == 10000
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cusparse_10_0.inc"
#elif CUDA_VERSION == 10010
#elif CUDA_VERSION < 10020
#include "tensorflow/stream_executor/cuda/cusparse_10_1.inc"
#elif CUDA_VERSION == 10020
#elif CUDA_VERSION < 11000
#include "tensorflow/stream_executor/cuda/cusparse_10_2.inc"
#elif CUSPARSE_VER_MAJOR == 11 && CUSPARSE_VER_MINOR == 0
#include "tensorflow/stream_executor/cuda/cusparse_11_0.inc"
#else
#error "We don't have a wrapper for this version."
#include "tensorflow/stream_executor/cuda/cusparse_11_0.inc"
#endif