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

Add 10.1 inc files for cuda libraries.

Browse Source 
The inc files for cufft, cupti and curand are identical to the ones for 10.0.

cuda_10_1.inc was not added since the stub file says "For now only one stub implementation is needed" and I don't know how to handle that correctly.

PiperOrigin-RevId: 264862947
This commit is contained in:
Guangda Lai 2019-08-22 10:18:40 -07:00 committed by TensorFlower Gardener
parent d285f3175a
commit bb403c3f8d
6 changed files with 16430 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5023
tensorflow/stream_executor/cuda/cublas_10_1.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -60,6 +60,8 @@ typedef enum {} cublasMath_t;
// Parameter constness changed in cuBLAS 9.2
#if CUDA_VERSION < 9020
#include "tensorflow/stream_executor/cuda/cublas_9_0.inc"
#else
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cublas_10_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cublas_10_1.inc"
#endif

3139
tensorflow/stream_executor/cuda/cusolver_dense_10_1.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -50,4 +50,8 @@ cusolverStatus_t GetSymbolNotFoundError() {
}
} // namespace
#if CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cusolver_dense_10_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cusolver_dense_10_1.inc"
#endif

8258
tensorflow/stream_executor/cuda/cusparse_10_1.inc Normal file
View File

File diff suppressed because it is too large Load Diff

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

@ -52,6 +52,8 @@ cusparseStatus_t GetSymbolNotFoundError() {
#if CUDA_VERSION < 9020
#include "tensorflow/stream_executor/cuda/cusparse_9_0.inc"
#else
#elif CUDA_VERSION < 10010
#include "tensorflow/stream_executor/cuda/cusparse_10_0.inc"
#else
#include "tensorflow/stream_executor/cuda/cusparse_10_1.inc"
#endif