Skip to content

Commit cfa7ffb

Browse files
A. Unique TensorFlowertensorflower-gardener
authored andcommitted
In the StreamExecutor, make lack of CUDA 7.5 a non-fatal error for SGEMM
(ie., propagate the error better). Change: 122297193
1 parent 85c0af6 commit cfa7ffb

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tensorflow/stream_executor/cuda/cuda_blas.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1685,8 +1685,9 @@ bool CUDABlas::DoBlasGemm(
16851685
&beta,
16861686
CUDAMemoryMutable(c), CUBLAS_DATA_HALF, ldc);
16871687
#else
1688-
LOG(FATAL) << "fp16 sgemm is not implemented in this cuBLAS version "
1688+
LOG(ERROR) << "fp16 sgemm is not implemented in this cuBLAS version "
16891689
<< "(need at least CUDA 7.5)";
1690+
return false;
16901691
#endif
16911692
}
16921693

0 commit comments

Comments
 (0)