hexu/tssc-hpcg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add check_cusparse

Browse Source
This commit is contained in:
abnerhexu
2026-01-18 22:45:26 +08:00
parent d7a4b39a6c
commit e0135503bc
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
DASP/src/main_spmv3_f64.cu
View File

@@ -46,15 +46,15 @@ void cusparse_spmv_all(MAT_VAL_TYPE *cu_ValA, MAT_PTR_TYPE *cu_RowPtrA, int *cu_
size_t bufferSize = 0;
gettimeofday(&t1, NULL);
cusparseCreate(&handle);
cusparseCreateCsr(&matA, rowA, colA, nnzA, dA_rpt, dA_cid, dA_val,
CHECK_CUSPARSEcusparseCreate(&handle));
CHECK_CUSPARSEcusparseCreateCsr(&matA, rowA, colA, nnzA, dA_rpt, dA_cid, dA_val,
CUSPARSE_INDEX_32I, CUSPARSE_INDEX_32I,
CUSPARSE_INDEX_BASE_ZERO, CUDA_R_64F);
cusparseCreateDnVec(&vecX, colA, dX, CUDA_R_64F);
cusparseCreateDnVec(&vecY, rowA, dY, CUDA_R_64F);
cusparseSpMV_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE,
CUSPARSE_INDEX_BASE_ZERO, CUDA_R_64F));
CHECK_CUSPARSE(cusparseCreateDnVec(&vecX, colA, dX, CUDA_R_64F));
CHECK_CUSPARSE(cusparseCreateDnVec(&vecY, rowA, dY, CUDA_R_64F));
CHECK_CUSPARSE(cusparseSpMV_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE,
&alpha, matA, vecX, &beta, vecY, CUDA_R_64F,
CUSPARSE_SPMV_ALG_DEFAULT, &bufferSize);
CUSPARSE_SPMV_ALG_DEFAULT, &bufferSize));
cudaMalloc(&dBuffer, bufferSize);
// cudaDeviceSynchronize();
gettimeofday(&t2, NULL);