Permalink
Browse files

Remove references missing GEMM algorithms

  • Loading branch information...
ljanyst committed May 20, 2017
1 parent c11537b commit cdf767190b00cbbabb96f1ab72a954bf47209ccc
Showing with 1 addition and 2 deletions.
  1. +1 −2 tensorflow/stream_executor/cuda/cuda_blas.cc
@@ -1912,8 +1912,7 @@ bool CUDABlas::GetBlasGemmAlgorithms(
#if CUDA_VERSION >= 8000
for (cublasGemmAlgo_t algo :
{CUBLAS_GEMM_DFALT, CUBLAS_GEMM_ALGO0, CUBLAS_GEMM_ALGO1,
CUBLAS_GEMM_ALGO2, CUBLAS_GEMM_ALGO3, CUBLAS_GEMM_ALGO4,
CUBLAS_GEMM_ALGO5, CUBLAS_GEMM_ALGO6, CUBLAS_GEMM_ALGO7}) {
CUBLAS_GEMM_ALGO2, CUBLAS_GEMM_ALGO3, CUBLAS_GEMM_ALGO4}) {
out_algorithms->push_back(algo);
}
#endif

0 comments on commit cdf7671

Please sign in to comment.