Skip to content

Commit e94960c

Browse files
committed
added missing mkl call in parcsrmv
1 parent 25f192c commit e94960c

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

SRC/parcsrmv.c

+2-2
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@ void pEVSL_ParcsrMatvecCommBegin(pevsl_Parcsr *A, double *x) {
3333
void pEVSL_ParcsrMatvecCommEnd(pevsl_Parcsr *A) {
3434
int err;
3535
err = MPI_Waitall(A->comm_handle->num_proc_send_to, A->comm_handle->send_requests, \
36-
A->comm_handle->send_status);
36+
A->comm_handle->send_status);
3737
PEVSL_CHKERR(err != MPI_SUCCESS);
3838
err = MPI_Waitall(A->comm_handle->num_proc_recv_from, A->comm_handle->recv_requests, \
39-
A->comm_handle->recv_status);
39+
A->comm_handle->recv_status);
4040
PEVSL_CHKERR(err != MPI_SUCCESS);
4141
}
4242

SRC/spmat.c

+7
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,14 @@ void dcsrgemv(char trans, int nrow, int ncol, double alp, double *a,
104104

105105
// y = alp*A*x + bet*y
106106
int pEVSL_MatvecGen(double alp, pevsl_Csr *A, double *x, double bet, double *y) {
107+
108+
#ifdef USE_MKL
109+
char cN = 'N';
110+
mkl_dcsrmv(&cN, &(A->nrows), &(A->ncols), &alp, "GXXCXX", A->a, A->ja, A->ia,
111+
A->ia+1, x, &bet, y);
112+
#else
107113
dcsrgemv('N', A->nrows, A->ncols, alp, A->a, A->ia, A->ja, x, bet, y);
114+
#endif
108115

109116
return 0;
110117
}

0 commit comments

Comments
 (0)