1 /* 2 Wrappers for mkl_cspblas_ routines. 3 A more elegant way to do this would be to use an approach like that used in petsclbaslapack_mangle.h, 4 but since the MKL sparse BLAS routines are not going to be as widely used, and because 5 we don't have to worry about Fortran name mangling, this seems OK for now. 6 */ 7 8 /* Have to redefine MKL_Complex16 and MKL_Complex8 as PetscScalar for the complex number cases. 9 * This works fine with a C99 compiler -- still need to verify that this works with C89. 10 * Note: These definitions need to occur BEFORE including MKL headers. */ 11 #define MKL_Complex16 PetscScalar 12 #define MKL_Complex8 PetscScalar 13 14 #if !defined(PETSC_USE_COMPLEX) 15 # if defined(PETSC_USE_REAL_SINGLE) 16 # define mkl_cspblas_xcsrgemv(transa,m,a,ia,ja,x,y) mkl_cspblas_scsrgemv(transa,m,a,ia,ja,x,y) 17 # elif defined(PETSC_USE_REAL_DOUBLE) 18 # define mkl_cspblas_xcsrgemv(transa,m,a,ia,ja,x,y) mkl_cspblas_dcsrgemv(transa,m,a,ia,ja,x,y) 19 # endif 20 #else 21 # if defined(PETSC_USE_REAL_SINGLE) 22 # define mkl_cspblas_xcsrgemv(transa,m,a,ia,ja,x,y) mkl_cspblas_ccsrgemv(transa,m,a,ia,ja,x,y) 23 # elif defined(PETSC_USE_REAL_DOUBLE) 24 # define mkl_cspblas_xcsrgemv(transa,m,a,ia,ja,x,y) mkl_cspblas_zcsrgemv(transa,m,a,ia,ja,x,y) 25 # endif 26 #endif 27 28 #if !defined(PETSC_USE_COMPLEX) 29 # if defined(PETSC_USE_REAL_SINGLE) 30 # define mkl_xcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) mkl_scsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) 31 # elif defined(PETSC_USE_REAL_DOUBLE) 32 # define mkl_xcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) mkl_dcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) 33 # endif 34 #else 35 # if defined(PETSC_USE_REAL_SINGLE) 36 # define mkl_xcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) mkl_ccsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) 37 # elif defined(PETSC_USE_REAL_DOUBLE) 38 # define mkl_xcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) mkl_zcsrmv(transa,m,k,alpha,matdescra,val,indx,pntrb,pntre,x,beta,y) 39 # endif 40 #endif 41 42 #if !defined(PETSC_USE_COMPLEX) 43 # if defined(PETSC_USE_REAL_SINGLE) 44 # define mkl_sparse_x_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) mkl_sparse_s_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) 45 # elif defined(PETSC_USE_REAL_DOUBLE) 46 # define mkl_sparse_x_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) mkl_sparse_d_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) 47 # endif 48 #else 49 # if defined(PETSC_USE_REAL_SINGLE) 50 # define mkl_sparse_x_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) mkl_sparse_c_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) 51 # elif defined(PETSC_USE_REAL_DOUBLE) 52 # define mkl_sparse_x_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) mkl_sparse_z_create_csr(A,indexing,rows,cols,rows_start,rows_end,col_indx,values) 53 # endif 54 #endif 55 56 #if !defined(PETSC_USE_COMPLEX) 57 # if defined(PETSC_USE_REAL_SINGLE) 58 # define mkl_sparse_x_mv(operation,alpha,A,descr,x,beta,y) mkl_sparse_s_mv(operation,alpha,A,descr,x,beta,y) 59 # elif defined(PETSC_USE_REAL_DOUBLE) 60 # define mkl_sparse_x_mv(operation,alpha,A,descr,x,beta,y) mkl_sparse_d_mv(operation,alpha,A,descr,x,beta,y) 61 # endif 62 #else 63 # if defined(PETSC_USE_REAL_SINGLE) 64 # define mkl_sparse_x_mv(operation,alpha,A,descr,x,beta,y) mkl_sparse_c_mv(operation,alpha,A,descr,x,beta,y) 65 # elif defined(PETSC_USE_REAL_DOUBLE) 66 # define mkl_sparse_x_mv(operation,alpha,A,descr,x,beta,y) mkl_sparse_z_mv(operation,alpha,A,descr,x,beta,y) 67 # endif 68 #endif 69