diff options
author | Justin Lebar <jlebar@google.com> | 2018-07-31 16:43:01 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-31 16:48:38 -0700 |
commit | 2bf582914d09207ad7276e2f471ea9776415e8e0 (patch) | |
tree | 210be03410eb5a115245228eeb54117f0f262b33 /tensorflow/stream_executor | |
parent | 28f0c8e37bd99b43d6d9597cf995f5f8e26f1056 (diff) |
[SE] Add new cublas algorithms from CUDA 9.2.
I verified that CUDA 9.1 did not introduce any new algorithms.
PiperOrigin-RevId: 206850523
Diffstat (limited to 'tensorflow/stream_executor')
-rw-r--r-- | tensorflow/stream_executor/cuda/cuda_blas.cc | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_blas.cc b/tensorflow/stream_executor/cuda/cuda_blas.cc index 67babd7f79..efba5be349 100644 --- a/tensorflow/stream_executor/cuda/cuda_blas.cc +++ b/tensorflow/stream_executor/cuda/cuda_blas.cc @@ -2257,6 +2257,25 @@ bool CUDABlas::GetBlasGemmAlgorithms( CUBLAS_GEMM_ALGO3_TENSOR_OP, CUBLAS_GEMM_ALGO4_TENSOR_OP, #endif +#if CUDA_VERSION >= 9200 + CUBLAS_GEMM_ALGO18, + CUBLAS_GEMM_ALGO19, + CUBLAS_GEMM_ALGO20, + CUBLAS_GEMM_ALGO21, + CUBLAS_GEMM_ALGO22, + CUBLAS_GEMM_ALGO23, + CUBLAS_GEMM_ALGO5_TENSOR_OP, + CUBLAS_GEMM_ALGO6_TENSOR_OP, + CUBLAS_GEMM_ALGO7_TENSOR_OP, + CUBLAS_GEMM_ALGO8_TENSOR_OP, + CUBLAS_GEMM_ALGO9_TENSOR_OP, + CUBLAS_GEMM_ALGO10_TENSOR_OP, + CUBLAS_GEMM_ALGO11_TENSOR_OP, + CUBLAS_GEMM_ALGO12_TENSOR_OP, + CUBLAS_GEMM_ALGO13_TENSOR_OP, + CUBLAS_GEMM_ALGO14_TENSOR_OP, + CUBLAS_GEMM_ALGO15_TENSOR_OP, +#endif }; return true; } |