extern "C" magma_int_t
magma_z_precond(
magma_z_sparse_matrix A,
magma_z_vector b,
magma_z_vector *x,
magma_z_preconditioner *precond,
magma_queue_t queue )
{
// set up precond parameters as solver parameters
magma_z_solver_par psolver_par;
psolver_par.epsilon = precond->epsilon;
psolver_par.maxiter = precond->maxiter;
psolver_par.restart = precond->restart;
psolver_par.verbose = 0;
switch( precond->solver ) {
case Magma_CG:
magma_zcg_res( A, b, x, &psolver_par, queue );break;
case Magma_BICGSTAB:
magma_zbicgstab( A, b, x, &psolver_par, queue );break;
case Magma_GMRES:
magma_zgmres( A, b, x, &psolver_par, queue );break;
case Magma_JACOBI:
magma_zjacobi( A, b, x, &psolver_par, queue );break;
case Magma_BAITER:
magma_zbaiter( A, b, x, &psolver_par, queue );break;
}
return MAGMA_SUCCESS;
}
/* ////////////////////////////////////////////////////////////////////////////
-- running magma_zbaiter
*/
int main( int argc, char** argv)
{
TESTING_INIT();
magma_z_solver_par solver_par;
magma_z_preconditioner precond_par;
solver_par.maxiter = 1000;
solver_par.verbose = 0;
solver_par.num_eigenvalues = 0;
int scale = 0;
magma_scale_t scaling = Magma_NOSCALE;
magma_z_sparse_matrix A;
magma_z_vector x, b;
magmaDoubleComplex one = MAGMA_Z_MAKE(1.0, 0.0);
magmaDoubleComplex zero = MAGMA_Z_MAKE(0.0, 0.0);
int i;
for( i = 1; i < argc; ++i ) {
if ( strcmp("--maxiter", argv[i]) == 0 ){
solver_par.maxiter = atoi( argv[++i] );
}else if ( strcmp("--mscale", argv[i]) == 0 ) {
scale = atoi( argv[++i] );
switch( scale ) {
case 0: scaling = Magma_NOSCALE; break;
case 1: scaling = Magma_UNITDIAG; break;
case 2: scaling = Magma_UNITROW; break;
}
}else
break;
}
printf( "\n# usage: ./run_zbaiter"
" [ "
" --mscale %d (0=no, 1=unitdiag, 2=unitrownrm)"
" --maxiter %d ]"
" matrices \n\n",
(int) scale,
(int) solver_par.maxiter);
magma_zsolverinfo_init( &solver_par, &precond_par );
while( i < argc ){
magma_z_csr_mtx( &A, argv[i] );
printf( "\n# matrix info: %d-by-%d with %d nonzeros\n\n",
(int) A.num_rows,(int) A.num_cols,(int) A.nnz );
// scale initial guess
magma_zmscale( &A, scaling );
magma_z_vinit( &b, Magma_DEV, A.num_cols, one );
magma_z_vinit( &x, Magma_DEV, A.num_cols, zero );
magma_zbaiter( A, b, &x, &solver_par );
magma_zsolverinfo( &solver_par, &precond_par );
magma_z_mfree(&A);
magma_z_vfree(&x);
magma_z_vfree(&b);
i++;
}
magma_zsolverinfo_free( &solver_par, &precond_par );
TESTING_FINALIZE();
return 0;
}
extern "C" magma_int_t
magma_z_solver(
magma_z_matrix A, magma_z_matrix b,
magma_z_matrix *x, magma_zopts *zopts,
magma_queue_t queue )
{
magma_int_t info = 0;
// make sure RHS is a dense matrix
if ( b.storage_type != Magma_DENSE ) {
printf( "error: sparse RHS not yet supported.\n" );
return MAGMA_ERR_NOT_SUPPORTED;
}
if( b.num_cols == 1 ){
switch( zopts->solver_par.solver ) {
case Magma_CG:
CHECK( magma_zcg_res( A, b, x, &zopts->solver_par, queue )); break;
case Magma_BICG:
CHECK( magma_zbicg( A, b, x, &zopts->solver_par, queue )); break;
case Magma_PBICG:
CHECK( magma_zpbicg( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_CGMERGE:
CHECK( magma_zcg_merge( A, b, x, &zopts->solver_par, queue )); break;
case Magma_PCG:
CHECK( magma_zpcg( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_PCGMERGE:
CHECK( magma_zpcg_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_BICGSTAB:
CHECK( magma_zbicgstab( A, b, x, &zopts->solver_par, queue )); break;
case Magma_BICGSTABMERGE:
CHECK( magma_zbicgstab_merge( A, b, x, &zopts->solver_par, queue )); break;
case Magma_PBICGSTABMERGE:
CHECK( magma_zpbicgstab_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_PBICGSTAB:
CHECK( magma_zpbicgstab( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_GMRES:
CHECK( magma_zfgmres( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_PGMRES:
CHECK( magma_zfgmres( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_IDR:
CHECK( magma_zidr( A, b, x, &zopts->solver_par, queue )); break;
case Magma_IDRMERGE:
CHECK( magma_zidr_merge( A, b, x, &zopts->solver_par, queue )); break;
//CHECK( magma_zidr_strms( A, b, x, &zopts->solver_par, queue )); break;
case Magma_PIDR:
CHECK( magma_zpidr( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_PIDRMERGE:
CHECK( magma_zpidr_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
//CHECK( magma_zpidr_strms( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_LOBPCG:
CHECK( magma_zlobpcg( A, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_ITERREF:
CHECK( magma_ziterref( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_JACOBI:
CHECK( magma_zftjacobi( A, b, x, &zopts->solver_par, queue )); break;
case Magma_BAITER:
CHECK( magma_zbaiter( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_BAITERO:
CHECK( magma_zbaiter_overlap( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_CGS:
CHECK( magma_zcgs( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_CGSMERGE:
CHECK( magma_zcgs_merge( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_PCGS:
CHECK( magma_zpcgs( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_PCGSMERGE:
CHECK( magma_zpcgs_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_TFQMR:
CHECK( magma_ztfqmr( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_PTFQMR:
CHECK( magma_zptfqmr( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_TFQMRMERGE:
CHECK( magma_ztfqmr_merge( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_PTFQMRMERGE:
CHECK( magma_zptfqmr_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_QMR:
CHECK( magma_zqmr( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_PQMR:
CHECK( magma_zpqmr( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_PQMRMERGE:
CHECK( magma_zpqmr_merge( A, b, x, &zopts->solver_par, &zopts->precond_par, queue ) ); break;
case Magma_QMRMERGE:
CHECK( magma_zqmr_merge( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_BOMBARD:
CHECK( magma_zbombard( A, b, x, &zopts->solver_par, queue ) ); break;
case Magma_BOMBARDMERGE:
CHECK( magma_zbombard_merge( A, b, x, &zopts->solver_par, queue ) ); break;
default:
printf("error: solver class not supported.\n"); break;
}
}
else {
switch( zopts->solver_par.solver ) {
case Magma_CG:
CHECK( magma_zbpcg( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_PCG:
CHECK( magma_zbpcg( A, b, x, &zopts->solver_par, &zopts->precond_par, queue )); break;
case Magma_LOBPCG:
CHECK( magma_zlobpcg( A, &zopts->solver_par, &zopts->precond_par, queue )); break;
default:
printf("error: only 1 RHS supported for this solver class.\n"); break;
}
}
cleanup:
return info;
}
