From 67c545fce0a1f40f12b13f2fa653b6899b60481e Mon Sep 17 00:00:00 2001 From: Ruipeng Li Date: Tue, 7 Jun 2022 15:28:11 -0700 Subject: [PATCH] bug fix matvecT --- src/parcsr_mv/_hypre_parcsr_mv.h | 4 ++-- src/parcsr_mv/par_csr_matop_device.c | 6 +++--- src/parcsr_mv/par_csr_matvec.c | 6 +++--- src/parcsr_mv/protos.h | 4 ++-- 4 files changed, 10 insertions(+), 10 deletions(-) diff --git a/src/parcsr_mv/_hypre_parcsr_mv.h b/src/parcsr_mv/_hypre_parcsr_mv.h index 4d4ea6a66..884be75ff 100644 --- a/src/parcsr_mv/_hypre_parcsr_mv.h +++ b/src/parcsr_mv/_hypre_parcsr_mv.h @@ -961,7 +961,7 @@ HYPRE_Int hypre_ParCSRMatrixDropSmallEntriesHost( hypre_ParCSRMatrix *A, HYPRE_R HYPRE_Int hypre_ParCSRMatrixDropSmallEntriesDevice( hypre_ParCSRMatrix *A, HYPRE_Complex tol, HYPRE_Int type); -HYPRE_Int hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, HYPRE_Int local_nrows ); +HYPRE_Int hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, HYPRE_Int local_ncols ); #ifdef HYPRE_USING_PERSISTENT_COMM hypre_ParCSRPersistentCommHandle* hypre_ParCSRPersistentCommHandleCreate(HYPRE_Int job, @@ -1074,7 +1074,7 @@ HYPRE_Int hypre_ParCSRMatrixMatvec ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A, HYPRE_Complex beta, hypre_ParVector *y ); HYPRE_Int hypre_ParCSRMatrixMatvecT ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A, hypre_ParVector *x, HYPRE_Complex beta, hypre_ParVector *y ); -HYPRE_Int hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int nrows, HYPRE_Complex *locl_data, +HYPRE_Int hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int ncols, HYPRE_Complex *locl_data, HYPRE_Complex *recv_data, hypre_ParCSRCommPkg *comm_pkg ); HYPRE_Int hypre_ParCSRMatrixMatvec_FF ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A, diff --git a/src/parcsr_mv/par_csr_matop_device.c b/src/parcsr_mv/par_csr_matop_device.c index 27dc52727..30398a4a3 100644 --- a/src/parcsr_mv/par_csr_matop_device.c +++ b/src/parcsr_mv/par_csr_matop_device.c @@ -943,13 +943,13 @@ hypre_ParcsrGetExternalRowsDeviceWait(void *vrequest) HYPRE_Int hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, - HYPRE_Int local_nrows ) + HYPRE_Int local_ncols ) { HYPRE_Int num_sends = hypre_ParCSRCommPkgNumSends(comm_pkg); HYPRE_Int num_elemt = hypre_ParCSRCommPkgSendMapStart(comm_pkg, num_sends); HYPRE_Int *send_map = hypre_ParCSRCommPkgDeviceSendMapElmts(comm_pkg); - hypre_CSRMatrix *matrix_E = hypre_CSRMatrixCreate(local_nrows, num_elemt, num_elemt); + hypre_CSRMatrix *matrix_E = hypre_CSRMatrixCreate(local_ncols, num_elemt, num_elemt); hypre_CSRMatrixMemoryLocation(matrix_E) = HYPRE_MEMORY_DEVICE; HYPRE_Int *e_ii = hypre_TAlloc(HYPRE_Int, num_elemt, HYPRE_MEMORY_DEVICE); @@ -960,7 +960,7 @@ hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, HYPRE_THRUST_CALL( sequence, e_j, e_j + num_elemt); HYPRE_THRUST_CALL( stable_sort_by_key, e_ii, e_ii + num_elemt, e_j ); - HYPRE_Int *e_i = hypreDevice_CsrRowIndicesToPtrs(local_nrows, num_elemt, e_ii); + HYPRE_Int *e_i = hypreDevice_CsrRowIndicesToPtrs(local_ncols, num_elemt, e_ii); HYPRE_Int *new_end = HYPRE_THRUST_CALL( unique, e_ii, e_ii + num_elemt); HYPRE_Int nid = new_end - e_ii; diff --git a/src/parcsr_mv/par_csr_matvec.c b/src/parcsr_mv/par_csr_matvec.c index 52311c25e..5da75d708 100644 --- a/src/parcsr_mv/par_csr_matvec.c +++ b/src/parcsr_mv/par_csr_matvec.c @@ -722,7 +722,7 @@ hypre_ParCSRMatrixMatvecT( HYPRE_Complex alpha, #if defined(HYPRE_USING_CUDA) || defined(HYPRE_USING_HIP) /* unpack recv data on device */ - hypre_ParCSRMatrixMatvecT_unpack( hypre_ParCSRMatrixNumRows(A), locl_data, recv_data, comm_pkg ); + hypre_ParCSRMatrixMatvecT_unpack( hypre_ParCSRMatrixNumCols(A), locl_data, recv_data, comm_pkg ); #elif defined(HYPRE_USING_DEVICE_OPENMP) HYPRE_Int i, j; /* unpack recv data on device */ @@ -922,7 +922,7 @@ hypre_ParCSRMatrixMatvec_FF( HYPRE_Complex alpha, #if defined(HYPRE_USING_CUDA) || defined(HYPRE_USING_HIP) HYPRE_Int -hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int nrows, +hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int ncols, HYPRE_Complex *locl_data, HYPRE_Complex *recv_data, hypre_ParCSRCommPkg *comm_pkg ) @@ -937,7 +937,7 @@ hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int nrows, if (!hypre_ParCSRCommPkgMatrixE(comm_pkg)) { - hypre_ParCSRCommPkgCreateMatrixE(comm_pkg, nrows); + hypre_ParCSRCommPkgCreateMatrixE(comm_pkg, ncols); } hypre_CSRMatrix *E = hypre_ParCSRCommPkgMatrixE(comm_pkg); diff --git a/src/parcsr_mv/protos.h b/src/parcsr_mv/protos.h index f85cbdc77..7b36356c5 100644 --- a/src/parcsr_mv/protos.h +++ b/src/parcsr_mv/protos.h @@ -336,7 +336,7 @@ HYPRE_Int hypre_ParCSRMatrixDropSmallEntriesHost( hypre_ParCSRMatrix *A, HYPRE_R HYPRE_Int hypre_ParCSRMatrixDropSmallEntriesDevice( hypre_ParCSRMatrix *A, HYPRE_Complex tol, HYPRE_Int type); -HYPRE_Int hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, HYPRE_Int local_nrows ); +HYPRE_Int hypre_ParCSRCommPkgCreateMatrixE( hypre_ParCSRCommPkg *comm_pkg, HYPRE_Int local_ncols ); #ifdef HYPRE_USING_PERSISTENT_COMM hypre_ParCSRPersistentCommHandle* hypre_ParCSRPersistentCommHandleCreate(HYPRE_Int job, @@ -449,7 +449,7 @@ HYPRE_Int hypre_ParCSRMatrixMatvec ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A, HYPRE_Complex beta, hypre_ParVector *y ); HYPRE_Int hypre_ParCSRMatrixMatvecT ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A, hypre_ParVector *x, HYPRE_Complex beta, hypre_ParVector *y ); -HYPRE_Int hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int nrows, HYPRE_Complex *locl_data, +HYPRE_Int hypre_ParCSRMatrixMatvecT_unpack( HYPRE_Int ncols, HYPRE_Complex *locl_data, HYPRE_Complex *recv_data, hypre_ParCSRCommPkg *comm_pkg ); HYPRE_Int hypre_ParCSRMatrixMatvec_FF ( HYPRE_Complex alpha, hypre_ParCSRMatrix *A,