CFD/hypre
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Parspgemm (#641)

Browse Source 
Merge two spgemm branches into one
This commit is contained in:
Ruipeng Li 2022-05-26 14:38:15 -07:00 committed by GitHub
parent 09eb54520c
commit 4ac9d00577
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 81 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/parcsr_ls/par_ilu_setup.c
View File

@ -1640,11 +1640,13 @@ HYPRE_ILUSetupCusparseCSRILU0SetupSolve(hypre_CSRMatrix *A, cusparseMatDescr_t m
HYPRE_CUSPARSE_CALL(cusparseCreateCsrsv2Info(&(matU_info)));
/* 2. Get working array size */
HYPRE_CUSPARSE_CALL(hypre_cusparse_csrsv2_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE, n, nnz_A,
HYPRE_CUSPARSE_CALL(hypre_cusparse_csrsv2_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE, n,
nnz_A,
matL_des, A_data, A_i, A_j,
matL_info, &matL_buffersize));
HYPRE_CUSPARSE_CALL(hypre_cusparse_csrsv2_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE, n, nnz_A,
HYPRE_CUSPARSE_CALL(hypre_cusparse_csrsv2_bufferSize(handle, CUSPARSE_OPERATION_NON_TRANSPOSE, n,
nnz_A,
matU_des, A_data, A_i, A_j,
matU_info, &matU_buffersize));

93
src/parcsr_mv/par_csr_triplemat_device.c
View File

@ -368,8 +368,25 @@ hypre_ParCSRTMatMatKTDevice( hypre_ParCSRMatrix *A,
#if PARCSRGEMM_TIMING > 1
t1 = hypre_MPI_Wtime();
#endif
hypre_CSRMatrixTranspose(A_diag, &AT_diag, 1);
hypre_CSRMatrixTranspose(A_offd, &AT_offd, 1);
if (hypre_ParCSRMatrixDiagT(A))
{
AT_diag = hypre_ParCSRMatrixDiagT(A);
}
else
{
hypre_CSRMatrixTranspose(A_diag, &AT_diag, 1);
}
if (hypre_ParCSRMatrixOffdT(A))
{
AT_offd = hypre_ParCSRMatrixOffdT(A);
}
else
{
hypre_CSRMatrixTranspose(A_offd, &AT_offd, 1);
}
#if PARCSRGEMM_TIMING > 1
hypre_ForceSyncComputeStream(hypre_handle());
t2 = hypre_MPI_Wtime() - t1;
@ -386,15 +403,28 @@ hypre_ParCSRTMatMatKTDevice( hypre_ParCSRMatrix *A,
hypre_ParPrintf(comm, "Time Stack %f\n", t2);
#endif
if (keep_transpose)
if (!hypre_ParCSRMatrixDiagT(A))
{
hypre_ParCSRMatrixDiagT(A) = AT_diag;
hypre_ParCSRMatrixOffdT(A) = AT_offd;
if (keep_transpose)
{
hypre_ParCSRMatrixDiagT(A) = AT_diag;
}
else
{
hypre_CSRMatrixDestroy(AT_diag);
}
}
else
if (!hypre_ParCSRMatrixOffdT(A))
{
hypre_CSRMatrixDestroy(AT_diag);
hypre_CSRMatrixDestroy(AT_offd);
if (keep_transpose)
{
hypre_ParCSRMatrixOffdT(A) = AT_offd;
}
else
{
hypre_CSRMatrixDestroy(AT_offd);
}
}
#if PARCSRGEMM_TIMING > 1
@ -632,19 +662,48 @@ hypre_ParCSRMatrixRAPKTDevice( hypre_ParCSRMatrix *R,
Abar = hypre_ConcatDiagAndOffdDevice(A);
hypre_CSRMatrixTransposeDevice(R_diag, &R_diagT, 1);
hypre_CSRMatrixTransposeDevice(R_offd, &R_offdT, 1);
RbarT = hypre_CSRMatrixStack2Device(R_diagT, R_offdT);
if (keep_transpose)
if (hypre_ParCSRMatrixDiagT(R))
{
hypre_ParCSRMatrixDiagT(R) = R_diagT;
hypre_ParCSRMatrixOffdT(R) = R_offdT;
R_diagT = hypre_ParCSRMatrixDiagT(R);
}
else
{
hypre_CSRMatrixDestroy(R_diagT);
hypre_CSRMatrixDestroy(R_offdT);
hypre_CSRMatrixTransposeDevice(R_diag, &R_diagT, 1);
}
if (hypre_ParCSRMatrixOffdT(R))
{
R_offdT = hypre_ParCSRMatrixOffdT(R);
}
else
{
hypre_CSRMatrixTransposeDevice(R_offd, &R_offdT, 1);
}
RbarT = hypre_CSRMatrixStack2Device(R_diagT, R_offdT);
if (!hypre_ParCSRMatrixDiagT(R))
{
if (keep_transpose)
{
hypre_ParCSRMatrixDiagT(R) = R_diagT;
}
else
{
hypre_CSRMatrixDestroy(R_diagT);
}
}
if (!hypre_ParCSRMatrixOffdT(R))
{
if (keep_transpose)
{
hypre_ParCSRMatrixOffdT(R) = R_offdT;
}
else
{
hypre_CSRMatrixDestroy(R_offdT);
}
}
Pext = hypre_ParCSRMatrixExtractBExtDeviceWait(request);

2
src/test/ij_mm.c
View File

@ -293,7 +293,7 @@ void runjob2( HYPRE_ParCSRMatrix parcsr_A,
//cudaProfilerStart();
}
parcsr_B = hypre_ParCSRTMatMat(parcsr_A, parcsr_A);
parcsr_B = hypre_ParCSRTMatMatKT(parcsr_A, parcsr_A, 0);
if (i == rep - 1)
{