From 2bf582914d09207ad7276e2f471ea9776415e8e0 Mon Sep 17 00:00:00 2001
From: Justin Lebar <jlebar@google.com>
Date: Tue, 31 Jul 2018 16:43:01 -0700
Subject: [SE] Add new cublas algorithms from CUDA 9.2.

I verified that CUDA 9.1 did not introduce any new algorithms.

PiperOrigin-RevId: 206850523
---
 tensorflow/stream_executor/cuda/cuda_blas.cc | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'tensorflow/stream_executor')

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
@@ -2256,6 +2256,25 @@ bool CUDABlas::GetBlasGemmAlgorithms(
     CUBLAS_GEMM_ALGO2_TENSOR_OP,
     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;
-- 
cgit v1.2.3