From 106766c1b68ae67b7731ae481fe7feecbb94974c Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 13 Jun 2018 12:14:34 -0700
Subject: Fix a build failure when cuda version is less than 9000.

PiperOrigin-RevId: 200432478
---
 tensorflow/stream_executor/cuda/cuda_blas.cc | 2 ++
 1 file changed, 2 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 92c1a5fc07..31e407f199 100644
--- a/tensorflow/stream_executor/cuda/cuda_blas.cc
+++ b/tensorflow/stream_executor/cuda/cuda_blas.cc
@@ -2183,10 +2183,12 @@ bool CUDABlas::DoBlasGemmWithAlgorithmImpl(
 
   // Return false if we might be hitting a cuBLAS bug that produces the wrong
   // result. See nvbugs/2156201, b/79126339.
+#if (CUDA_VERSION >= 9000)
   if (CUDA_VERSION < 9020 && algorithm != CUBLAS_GEMM_ALGO12 &&
       std::max({m, n, k}) >= 2097153 && cc_major < 7) {
     return false;
   }
+#endif
 
   cudaDataType_t cuda_in_type = CUDADataType<InT>::type;
   // Since we are converting 'algorithm' to cublasGemmAlgo_t by static_cast,
-- 
cgit v1.2.3