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

Merge pull request #40393 from nluehr:stub_cleanup

Browse Source 
PiperOrigin-RevId: 316982263
Change-Id: Ic6aebd2137c192d186ccbb0f59e5472cd7062cab
This commit is contained in:
TensorFlower Gardener 2020-06-17 16:21:58 -07:00
commit d5163e15c2
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 #if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cublas_9_0.inc" #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" #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" #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" #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 #else
#error "We have no wrapper for this version." #include "tensorflow/stream_executor/cuda/cublas_11_0.inc"
#endif #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 #if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cuda_9_0.inc" #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" #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" #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" #include "tensorflow/stream_executor/cuda/cuda_10_2.inc"
#elif CUDA_VERSION <= 11000
#include "tensorflow/stream_executor/cuda/cuda_11_0.inc"
#else #else
#error "We have no wrapper for this version." #include "tensorflow/stream_executor/cuda/cuda_11_0.inc"
#endif #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 // A bunch of new symbols were introduced in version 10
#if CUDART_VERSION < 10000 #if CUDART_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cuda_runtime_9_0.inc" #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" #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" #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" #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 #else
#error "We have no wrapper for this version." #include "tensorflow/stream_executor/cuda/cuda_runtime_11_0.inc"
#endif #endif
#undef __dv #undef __dv
#undef __CUDA_DEPRECATED #undef __CUDA_DEPRECATED

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

@ -53,14 +53,12 @@ cusolverStatus_t GetSymbolNotFoundError() {
#if CUDA_VERSION < 10000 #if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cusolver_dense_9_0.inc" #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" #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" #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" #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 #else
#error "We don't have a wrapper for this version." #include "tensorflow/stream_executor/cuda/cusolver_dense_11_0.inc"
#endif #endif

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

@ -53,14 +53,12 @@ cusparseStatus_t GetSymbolNotFoundError() {
#if CUDA_VERSION < 10000 #if CUDA_VERSION < 10000
#include "tensorflow/stream_executor/cuda/cusparse_9_0.inc" #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" #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" #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" #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 #else
#error "We don't have a wrapper for this version." #include "tensorflow/stream_executor/cuda/cusparse_11_0.inc"
#endif #endif