OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix build failure with CUDA 11.0.0.

Browse Source 
undefined cusparseSparseToDenseAlg_t and cusparseDenseToSparseAlg_t

PiperOrigin-RevId: 372993274
Change-Id: I7ddea28644732ef6d5541644c27fe7f2abff7c0e
This commit is contained in:
Qiao Zhang
2021-05-10 12:56:27 -07:00
committed by TensorFlower Gardener
parent b362c92fcc
commit bf5b1056fc
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/stream_executor/cuda/cusparse_11_0.inc
View File

@@ -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,