diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-13 12:14:34 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-13 12:17:45 -0700 |
commit | 106766c1b68ae67b7731ae481fe7feecbb94974c (patch) | |
tree | c39ff28c2635d725fa75b1231c9e68358ad9cdd4 /tensorflow/stream_executor/cuda | |
parent | 74655a96b40680b111ae063386c57f3f38262d34 (diff) |
Fix a build failure when cuda version is less than 9000.
PiperOrigin-RevId: 200432478
Diffstat (limited to 'tensorflow/stream_executor/cuda')
-rw-r--r-- | tensorflow/stream_executor/cuda/cuda_blas.cc | 2 |
1 files changed, 2 insertions, 0 deletions
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, |