From bf5b1056fc3f77bbbebedbdba1dbe8480142db40 Mon Sep 17 00:00:00 2001 From: Qiao Zhang Date: Mon, 10 May 2021 12:56:27 -0700 Subject: [PATCH] Fix build failure with CUDA 11.0.0. undefined cusparseSparseToDenseAlg_t and cusparseDenseToSparseAlg_t PiperOrigin-RevId: 372993274 Change-Id: I7ddea28644732ef6d5541644c27fe7f2abff7c0e --- tensorflow/stream_executor/cuda/cusparse_11_0.inc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/stream_executor/cuda/cusparse_11_0.inc b/tensorflow/stream_executor/cuda/cusparse_11_0.inc index b8a13fb1dfa..f762efb1932 100644 --- a/tensorflow/stream_executor/cuda/cusparse_11_0.inc +++ b/tensorflow/stream_executor/cuda/cusparse_11_0.inc @@ -6671,6 +6671,8 @@ cusparseSpVV(cusparseHandle_t handle, cusparseOperation_t opX, return func_ptr(handle, opX, vecX, vecY, result, computeType, externalBuffer); } +#if CUSPARSE_VERSION >= 11300 + cusparseStatus_t CUSPARSEAPI cusparseSparseToDense_bufferSize( cusparseHandle_t handle, cusparseSpMatDescr_t matA, cusparseDnMatDescr_t matB, cusparseSparseToDenseAlg_t alg, @@ -6730,6 +6732,8 @@ cusparseStatus_t CUSPARSEAPI cusparseDenseToSparse_convert( return func_ptr(handle, matA, matB, alg, buffer); } +#endif // CUSPARSE_VERSION >= 11300 + cusparseStatus_t CUSPARSEAPI cusparseSpMV( cusparseHandle_t handle, cusparseOperation_t opA, const void *alpha, cusparseSpMatDescr_t matA, cusparseDnVecDescr_t vecX, const void *beta,