LIS_INT lis_cgs_switch(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, p,phat, q, qhat, u, uhat, vhat;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1, one;
LIS_REAL bnrm2, nrm2, tol, tol2;
LIS_INT iter,maxiter,n,output,conv;
LIS_INT iter2,maxiter2;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
maxiter2 = solver->options[LIS_OPTIONS_SWITCH_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
tol = solver->params[LIS_PARAMS_RESID-LIS_OPTIONS_LEN];
tol2 = solver->params[LIS_PARAMS_SWITCH_RESID-LIS_OPTIONS_LEN];
ptimes = 0.0;
r = solver->work[0];
rtld = solver->work[1];
p = solver->work[2];
phat = solver->work[3];
q = solver->work[4];
qhat = solver->work[5];
u = solver->work[5];
uhat = solver->work[6];
vhat = solver->work[6];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
LIS_QUAD_SCALAR_MALLOC(one,6,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
alpha.hi[0] = 1.0;
alpha.lo[0] = 0.0;
one.hi[0] = 1.0;
one.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol2 = solver->tol_switch;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, q);
lis_vector_set_allex_nm(0.0, p);
uhat->precision = LIS_PRECISION_DEFAULT;
p->precision = LIS_PRECISION_DEFAULT;
phat->precision = LIS_PRECISION_DEFAULT;
for( iter=1; iter<=maxiter2; iter++ )
{
/* rho = <rtld,r> */
lis_vector_dot(rtld,r,&rho.hi[0]);
/* test breakdown */
if( rho.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
beta.hi[0] = (rho.hi[0] / rho_old.hi[0]);
/* u = r + beta*q */
lis_vector_axpyz(beta.hi[0],q,r,u);
/* p = u + beta*(q + beta*p) */
lis_vector_xpay(q,beta.hi[0],p);
lis_vector_xpay(u,beta.hi[0],p);
/* phat = M^-1 * p */
times = lis_wtime();
lis_psolve(solver, p, phat);
ptimes += lis_wtime()-times;
/* v = A * phat */
LIS_MATVEC(A,phat,vhat);
/* tmpdot1 = <rtld,vhat> */
lis_vector_dot(rtld,vhat,&tmpdot1.hi[0]);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
alpha.hi[0] = rho.hi[0] / tmpdot1.hi[0];
/* q = u - alpha*vhat */
lis_vector_axpyz(-alpha.hi[0],vhat,u,q);
/* phat = u + q */
/* uhat = M^-1 * (u + q) */
lis_vector_axpyz(1.0,u,q,phat);
times = lis_wtime();
lis_psolve(solver, phat, uhat);
ptimes += lis_wtime()-times;
/* x = x + alpha*uhat */
lis_vector_axpy(alpha.hi[0],uhat,x);
/* qhat = A * uhat */
LIS_MATVEC(A,uhat,qhat);
/* r = r - alpha*qhat */
lis_vector_axpy(-alpha.hi[0],qhat,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
if( nrm2 <= tol2 )
{
solver->iter = iter;
solver->iter2 = iter;
solver->ptimes = ptimes;
break;
}
rho_old.hi[0] = rho.hi[0];
}
uhat->precision = LIS_PRECISION_QUAD;
p->precision = LIS_PRECISION_QUAD;
phat->precision = LIS_PRECISION_QUAD;
solver->options[LIS_OPTIONS_INITGUESS_ZEROS] = LIS_FALSE;
lis_vector_copyex_mn(x,solver->xx);
rho_old.hi[0] = 1.0;
lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2);
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, q);
lis_vector_set_allex_nm(0.0, p);
for( iter2=iter+1; iter2<=maxiter; iter2++ )
{
/* rho = <rtld,r> */
lis_vector_dotex_mmm(rtld,r,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
/* u = r + beta*q */
lis_vector_axpyzex_mmmm(beta,q,r,u);
/* p = u + beta*(q + beta*p) */
lis_vector_xpayex_mmm(q,beta,p);
lis_vector_xpayex_mmm(u,beta,p);
/* phat = M^-1 * p */
times = lis_wtime();
lis_psolve(solver, p, phat);
ptimes += lis_wtime()-times;
/* v = A * phat */
LIS_MATVEC(A,phat,vhat);
/* tmpdot1 = <rtld,vhat> */
lis_vector_dotex_mmm(rtld,vhat,&tmpdot1);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
/* q = u - alpha*vhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyzex_mmmm(alpha,vhat,u,q);
/* phat = u + q */
/* uhat = M^-1 * (u + q) */
lis_vector_axpyzex_mmmm(one,u,q,phat);
times = lis_wtime();
lis_psolve(solver, phat, uhat);
ptimes += lis_wtime()-times;
/* x = x + alpha*uhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,uhat,x);
/* qhat = A * uhat */
LIS_MATVEC(A,uhat,qhat);
/* r = r - alpha*qhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,qhat,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter2] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter2, nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_crs_quad(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, p, q, u, z, ap, map, uq, auq;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1, one;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,n,output,conv;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptimes = 0.0;
r = solver->work[0];
rtld = solver->work[1];
p = solver->work[2];
z = solver->work[3];
u = solver->work[3];
uq = solver->work[3];
q = solver->work[4];
ap = solver->work[4];
map = solver->work[5];
auq = solver->work[5];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
LIS_QUAD_SCALAR_MALLOC(one,6,1);
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,p);
LIS_MATVECT(A,p,rtld);
lis_vector_set_allex_nm(0.0,q);
lis_vector_set_allex_nm(0.0,p);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
one.hi[0] = 1.0;
one.lo[0] = 0.0;
for( iter=1; iter<=maxiter; iter++ )
{
/* z = M^-1 * r */
/* rho = <rtld,z> */
times = lis_wtime();
lis_psolve(solver, r, z);
ptimes += lis_wtime()-times;
lis_vector_dotex_mmm(rtld,z,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = rho / rho_old */
/* u = z + beta*q */
/* p = u + beta*(q + beta*p) */
/* ap = A * p */
/* map = M^-1 * ap */
/* tmpdot1 = <rtld,map> */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_vector_axpyzex_mmmm(beta,q,z,u);
lis_vector_xpayex_mmm(q,beta,p);
lis_vector_xpayex_mmm(u,beta,p);
LIS_MATVEC(A,p,ap);
times = lis_wtime();
lis_psolve(solver, ap, map);
ptimes += lis_wtime()-times;
lis_vector_dotex_mmm(rtld,map,&tmpdot1);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
/* q = u - alpha*map */
/* uq = u + q */
/* auq = A * uq */
/* x = x + alpha*uq */
/* r = r - alpha*auq */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyzex_mmmm(alpha,map,u,q);
lis_vector_axpyzex_mmmm(one,u,q,uq);
LIS_MATVEC(A,uq,auq);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,uq,x);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,auq,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
if( tol >= nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicr_quad(LIS_SOLVER solver)
{
LIS_MATRIX A,At;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, z,ztld,p, ptld, ap, map, az, aptld;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,n,output,conv;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
At = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptimes = 0.0;
r = solver->work[0];
rtld = solver->work[1];
z = solver->work[2];
ztld = solver->work[3];
p = solver->work[4];
ptld = solver->work[5];
ap = solver->work[6];
az = solver->work[7];
map = solver->work[8];
aptld = solver->work[9];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_psolve(solver, r, z);
lis_psolvet(solver, rtld, ztld);
lis_vector_copyex_mm(z,p);
lis_vector_copyex_mm(ztld,ptld);
LIS_MATVEC(A,z,ap);
lis_vector_dotex_mmm(ap,ztld,&rho_old);
for( iter=1; iter<=maxiter; iter++ )
{
/* aptld = A^T * ptld */
/* map = M^-1 * ap */
LIS_MATVECT(A,ptld,aptld);
times = lis_wtime();
lis_psolve(solver, ap, map);
ptimes += lis_wtime()-times;
/* tmpdot1 = <map,aptld> */
lis_vector_dotex_mmm(map,aptld,&tmpdot1);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho_old / tmpdot1 */
/* x = x + alpha*p */
/* r = r - alpha*ap */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho_old.hi,(LIS_QUAD *)tmpdot1.hi);
lis_vector_axpyex_mmm(alpha,p,x);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,ap,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol >= nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* rtld = rtld - alpha*aptld */
/* z = z - alpha*map */
/* ztld = M^-T * rtld */
/* az = A * z */
/* rho = <az,ztld> */
lis_vector_axpyex_mmm(alpha,aptld,rtld);
lis_vector_axpyex_mmm(alpha,map,z);
times = lis_wtime();
lis_psolvet(solver, rtld, ztld);
ptimes += lis_wtime()-times;
LIS_MATVEC(A,z,az);
lis_vector_dotex_mmm(az,ztld,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = rho / rho_old */
/* p = z + beta*p */
/* ptld = ztld + beta*ptld */
/* ap = az + beta*ap */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_vector_xpayex_mmm(z,beta,p);
lis_vector_xpayex_mmm(ztld,beta,ptld);
lis_vector_xpayex_mmm(az,beta,ap);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_cgs_quad(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_VECTOR x;
LIS_VECTOR r,rtld, p,phat, q, qhat, u, uhat, vhat;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1, one;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,output,conv;
double time,ptime;
LIS_DEBUG_FUNC_IN;
A = solver->A;
x = solver->x;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptime = 0.0;
r = solver->work[0];
rtld = solver->work[1];
p = solver->work[2];
phat = solver->work[3];
q = solver->work[4];
qhat = solver->work[5];
u = solver->work[5];
uhat = solver->work[6];
vhat = solver->work[6];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
LIS_QUAD_SCALAR_MALLOC(one,6,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
alpha.hi[0] = 1.0;
alpha.lo[0] = 0.0;
one.hi[0] = 1.0;
one.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, q);
lis_vector_set_allex_nm(0.0, p);
for( iter=1; iter<=maxiter; iter++ )
{
/* rho = <rtld,r> */
lis_vector_dotex_mmm(rtld,r,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
/* u = r + beta*q */
lis_vector_axpyzex_mmmm(beta,q,r,u);
/* p = u + beta*(q + beta*p) */
lis_vector_xpayex_mmm(q,beta,p);
lis_vector_xpayex_mmm(u,beta,p);
/* phat = M^-1 * p */
time = lis_wtime();
lis_psolve(solver, p, phat);
ptime += lis_wtime()-time;
/* v = A * phat */
lis_matvec(A,phat,vhat);
/* tmpdot1 = <rtld,vhat> */
lis_vector_dotex_mmm(rtld,vhat,&tmpdot1);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
/* q = u - alpha*vhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyzex_mmmm(alpha,vhat,u,q);
/* phat = u + q */
/* uhat = M^-1 * (u + q) */
lis_vector_axpyzex_mmmm(one,u,q,phat);
time = lis_wtime();
lis_psolve(solver, phat, uhat);
ptime += lis_wtime()-time;
/* x = x + alpha*uhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,uhat,x);
/* qhat = A * uhat */
lis_matvec(A,uhat,qhat);
/* r = r - alpha*qhat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,qhat,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptime = ptime;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicgstab_switch(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, t,p,v, s, phat, shat;
LIS_QUAD_PTR alpha, beta, omega, rho, rho_old, tmpdot1, tmpdot2;
LIS_REAL bnrm2, nrm2, tol, tol2;
LIS_INT iter,maxiter,n,output,conv;
LIS_INT iter2,maxiter2;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
maxiter2 = solver->options[LIS_OPTIONS_SWITCH_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
tol = solver->params[LIS_PARAMS_RESID-LIS_OPTIONS_LEN];
tol2 = solver->params[LIS_PARAMS_SWITCH_RESID-LIS_OPTIONS_LEN];
ptimes = 0.0;
rtld = solver->work[0];
r = solver->work[1];
s = solver->work[1];
t = solver->work[2];
p = solver->work[3];
v = solver->work[4];
phat = solver->work[5];
shat = solver->work[6];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
LIS_QUAD_SCALAR_MALLOC(omega,6,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot2,7,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
alpha.hi[0] = 1.0;
alpha.lo[0] = 0.0;
omega.hi[0] = 1.0;
omega.lo[0] = 0.0;
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, phat);
lis_vector_set_allex_nm(0.0, s);
lis_vector_set_allex_nm(0.0, shat);
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol2 = solver->tol_switch;
lis_solver_set_shadowresidual(solver,r,rtld);
s->precision = LIS_PRECISION_DEFAULT;
shat->precision = LIS_PRECISION_DEFAULT;
p->precision = LIS_PRECISION_DEFAULT;
phat->precision = LIS_PRECISION_DEFAULT;
for( iter=1; iter<=maxiter2; iter++ )
{
/* rho = <rtld,r> */
lis_vector_dot(rtld,r,&rho.hi[0]);
/* test breakdown */
if( rho.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
if( iter==1 )
{
lis_vector_copy(r,p);
}
else
{
/* beta = (rho / rho_old) * (alpha / omega) */
beta.hi[0] = (rho.hi[0] / rho_old.hi[0]) * (alpha.hi[0] / omega.hi[0]);
/* p = r + beta*(p - omega*v) */
lis_vector_axpy(-omega.hi[0],v,p);
lis_vector_xpay(r,beta.hi[0],p);
}
/* phat = M^-1 * p */
times = lis_wtime();
lis_psolve(solver, p, phat);
ptimes += lis_wtime()-times;
/* v = A * phat */
LIS_MATVEC(A,phat,v);
/* tmpdot1 = <rtld,v> */
lis_vector_dot(rtld,v,&tmpdot1.hi[0]);
/* test breakdown */
/* */
/* alpha = rho / tmpdot1 */
alpha.hi[0] = rho.hi[0] / tmpdot1.hi[0];
/* s = r - alpha*v */
lis_vector_axpy(-alpha.hi[0],v,r);
/* Early check for tolerance */
lis_solver_get_residual[conv](s,solver,&nrm2);
if( nrm2 <= tol2 )
{
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
lis_vector_axpy(alpha.hi[0],phat,x);
solver->iter = iter;
solver->iter2 = iter;
solver->ptimes = ptimes;
break;
}
/* shat = M^-1 * s */
times = lis_wtime();
lis_psolve(solver, s, shat);
ptimes += lis_wtime()-times;
/* t = A * shat */
LIS_MATVEC(A,shat,t);
/* tmpdot1 = <t,s> */
/* tmpdot2 = <t,t> */
/* omega = tmpdot1 / tmpdot2 */
lis_vector_dot(t,s,&tmpdot1.hi[0]);
lis_vector_dot(t,t,&tmpdot2.hi[0]);
omega.hi[0] = tmpdot1.hi[0] / tmpdot2.hi[0];
/* x = x + alpha*phat + omega*shat */
lis_vector_axpy(alpha.hi[0],phat,x);
lis_vector_axpy(omega.hi[0],shat,x);
/* r = s - omega*t */
lis_vector_axpy(-omega.hi[0],t,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
if( nrm2 <= tol2 )
{
solver->iter = iter;
solver->iter2 = iter;
solver->ptimes = ptimes;
break;
}
if( omega.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
rho_old.hi[0] = rho.hi[0];
}
s->precision = LIS_PRECISION_QUAD;
shat->precision = LIS_PRECISION_QUAD;
p->precision = LIS_PRECISION_QUAD;
phat->precision = LIS_PRECISION_QUAD;
solver->options[LIS_OPTIONS_INITGUESS_ZEROS] = LIS_FALSE;
lis_vector_copyex_mn(x,solver->xx);
rho_old.hi[0] = 1.0;
alpha.hi[0] = 1.0;
omega.hi[0] = 1.0;
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, phat);
lis_vector_set_allex_nm(0.0, s);
lis_vector_set_allex_nm(0.0, shat);
/* Initial Residual */
lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2);
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
for( iter2=iter+1; iter2<=maxiter; iter2++ )
{
/* rho = <rtld,r> */
lis_vector_dotex_mmm(rtld,r,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
if( iter2==1 )
{
lis_vector_copyex_mm(r,p);
}
else
{
/* beta = (rho / rho_old) * (alpha / omega) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_quad_div((LIS_QUAD *)tmpdot1.hi,(LIS_QUAD *)alpha.hi,(LIS_QUAD *)omega.hi);
lis_quad_mul((LIS_QUAD *)beta.hi,(LIS_QUAD *)beta.hi,(LIS_QUAD *)tmpdot1.hi);
/* p = r + beta*(p - omega*v) */
lis_quad_minus((LIS_QUAD *)omega.hi);
lis_vector_axpyex_mmm(omega,v,p);
lis_vector_xpayex_mmm(r,beta,p);
}
/* phat = M^-1 * p */
times = lis_wtime();
lis_psolve(solver, p, phat);
ptimes += lis_wtime()-times;
/* v = A * phat */
LIS_MATVEC(A,phat,v);
/* tmpdot1 = <rtld,v> */
lis_vector_dotex_mmm(rtld,v,&tmpdot1);
/* test breakdown */
/* */
/* alpha = rho / tmpdot1 */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
/* s = r - alpha*v */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,v,r);
/* Early check for tolerance */
lis_solver_get_residual[conv](s,solver,&nrm2);
if( tol > nrm2 )
{
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter2] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter2, nrm2);
}
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,phat,x);
solver->retcode = LIS_SUCCESS;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* shat = M^-1 * s */
times = lis_wtime();
lis_psolve(solver, s, shat);
ptimes += lis_wtime()-times;
/* t = A * shat */
LIS_MATVEC(A,shat,t);
/* tmpdot1 = <t,s> */
/* tmpdot2 = <t,t> */
/* omega = tmpdot1 / tmpdot2 */
lis_vector_dotex_mmm(t,s,&tmpdot1);
lis_vector_dotex_mmm(t,t,&tmpdot2);
lis_quad_div((LIS_QUAD *)omega.hi,(LIS_QUAD *)tmpdot1.hi,(LIS_QUAD *)tmpdot2.hi);
/* x = x + alpha*phat + omega*shat */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,phat,x);
lis_vector_axpyex_mmm(omega,shat,x);
/* r = s - omega*t */
lis_quad_minus((LIS_QUAD *)omega.hi);
lis_vector_axpyex_mmm(omega,t,r);
lis_quad_minus((LIS_QUAD *)omega.hi);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter2] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter2, nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
if( omega.hi[0]==0.0 && omega.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicg_switch(LIS_SOLVER solver)
{
LIS_MATRIX A,At;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, z,ztld,p, ptld, q, qtld;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1;
LIS_REAL bnrm2, nrm2, tol, tol2;
LIS_INT iter,maxiter,n,output,conv;
LIS_INT iter2,maxiter2;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
At = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
maxiter2 = solver->options[LIS_OPTIONS_SWITCH_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
tol = solver->params[LIS_PARAMS_RESID-LIS_OPTIONS_LEN];
tol2 = solver->params[LIS_PARAMS_SWITCH_RESID-LIS_OPTIONS_LEN];
ptimes = 0.0;
r = solver->work[0];
rtld = solver->work[1];
z = solver->work[2];
ztld = solver->work[3];
p = solver->work[4];
ptld = solver->work[5];
q = solver->work[2];
qtld = solver->work[3];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol2 = solver->tol_switch;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, ptld);
r->precision = LIS_PRECISION_DEFAULT;
rtld->precision = LIS_PRECISION_DEFAULT;
p->precision = LIS_PRECISION_DEFAULT;
ptld->precision = LIS_PRECISION_DEFAULT;
for( iter=1; iter<=maxiter2; iter++ )
{
/* z = M^-1 * r */
/* ztld = M^-T * rtld */
times = lis_wtime();
lis_psolve(solver, r, z);
lis_psolvet(solver, rtld, ztld);
ptimes += lis_wtime()-times;
/* rho = <z,rtld> */
lis_vector_dot(z,rtld,&rho.hi[0]);
/* test breakdown */
if( rho.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
beta.hi[0] = rho.hi[0] / rho_old.hi[0];
/* p = z + beta*p */
/* ptld = ztld + beta*ptld */
/* q = A * p */
/* qtld = A^T * ptld */
lis_vector_xpay(z,beta.hi[0],p);
LIS_MATVEC(A,p,q);
lis_vector_xpay(ztld,beta.hi[0],ptld);
LIS_MATVECT(At,ptld,qtld);
/* tmpdot1 = <ptld,q> */
lis_vector_dot(ptld,q,&tmpdot1.hi[0]);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
alpha.hi[0] = rho.hi[0] / tmpdot1.hi[0];
/* x = x + alpha*p */
lis_vector_axpy(alpha.hi[0],p,x);
/* r = r - alpha*q */
lis_vector_axpy(-alpha.hi[0],q,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( nrm2 <= tol2 )
{
solver->iter = iter;
solver->iter2 = iter;
solver->ptimes = ptimes;
break;
}
/* rtld = rtld - alpha*qtld */
lis_vector_axpy(-alpha.hi[0],qtld,rtld);
rho_old.hi[0] = rho.hi[0];
}
r->precision = LIS_PRECISION_QUAD;
rtld->precision = LIS_PRECISION_QUAD;
p->precision = LIS_PRECISION_QUAD;
ptld->precision = LIS_PRECISION_QUAD;
/* solver->precon->precon_type = 0;*/
solver->options[LIS_OPTIONS_INITGUESS_ZEROS] = LIS_FALSE;
lis_vector_copyex_mn(x,solver->xx);
rho_old.hi[0] = 1.0;
lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2);
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, ptld);
for( iter2=iter+1; iter2<=maxiter; iter2++ )
{
/* z = M^-1 * r */
/* ztld = M^-T * rtld */
times = lis_wtime();
lis_psolve(solver, r, z);
lis_psolvet(solver, rtld, ztld);
/* memset(z->value_lo,0,n*sizeof(LIS_SCALAR));
memset(ztld->value_lo,0,n*sizeof(LIS_SCALAR));*/
ptimes += lis_wtime()-times;
/* rho = <z,rtld> */
lis_vector_dotex_mmm(z,rtld,&rho);
/* printf("rho = %e %e\n",rho.hi[0],rho.lo[0]);*/
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
/* p = z + beta*p */
/* ptld = ztld + beta*ptld */
/* q = A * p */
/* qtld = A^T * ptld */
lis_vector_xpayex_mmm(z,beta,p);
LIS_MATVEC(A,p,q);
lis_vector_xpayex_mmm(ztld,beta,ptld);
LIS_MATVECT(At,ptld,qtld);
/* tmpdot1 = <ptld,q> */
lis_vector_dotex_mmm(ptld,q,&tmpdot1);
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
/* x = x + alpha*p */
lis_vector_axpyex_mmm(alpha,p,x);
/* r = r - alpha*q */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,q,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter2] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* rtld = rtld - alpha*qtld */
lis_vector_axpyex_mmm(alpha,qtld,rtld);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicrstab_quad(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_PRECON M;
LIS_VECTOR b,x;
LIS_VECTOR r,rtld, p, s, ap, ms, map, ams, z;
LIS_QUAD_PTR alpha, beta, omega, rho, rho_old, tmpdot1, tmpdot2;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,n,output,conv;
double times,ptimes;
LIS_DEBUG_FUNC_IN;
A = solver->A;
M = solver->precon;
b = solver->b;
x = solver->x;
n = A->n;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptimes = 0.0;
rtld = solver->work[0];
r = solver->work[1];
s = solver->work[2];
ms = solver->work[3];
ams = solver->work[4];
p = solver->work[5];
ap = solver->work[6];
map = solver->work[7];
z = solver->work[8];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
LIS_QUAD_SCALAR_MALLOC(omega,6,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot2,7,1);
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,p);
LIS_MATVECT(A,p,rtld);
times = lis_wtime();
lis_psolve(solver, r, z);
ptimes += lis_wtime()-times;
lis_vector_copyex_mm(z,p);
lis_vector_dotex_mmm(rtld,z,&rho_old);
for( iter=1; iter<=maxiter; iter++ )
{
/* ap = A * p */
/* map = M^-1 * ap */
/* tmpdot1 = <rtld,map> */
/* alpha = rho_old / tmpdot1 */
/* s = r - alpha*ap */
LIS_MATVEC(A,p,ap);
times = lis_wtime();
lis_psolve(solver, ap, map);
ptimes += lis_wtime()-times;
lis_vector_dotex_mmm(rtld,map,&tmpdot1);
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho_old.hi,(LIS_QUAD *)tmpdot1.hi);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyzex_mmmm(alpha,ap,r,s);
/* Early check for tolerance */
lis_solver_get_residual[conv](s,solver,&nrm2);
if( nrm2 <= tol )
{
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,p,x);
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* ms = z - alpha*map */
/* ams = A * ms */
/* tmpdot1 = <ams,s> */
/* tmpdot2 = <ams,ams> */
/* omega = tmpdot1 / tmpdot2 */
lis_vector_axpyzex_mmmm(alpha,map,z,ms);
LIS_MATVEC(A,ms,ams);
lis_vector_dotex_mmm(ams,s,&tmpdot1);
lis_vector_dotex_mmm(ams,ams,&tmpdot2);
lis_quad_div((LIS_QUAD *)omega.hi,(LIS_QUAD *)tmpdot1.hi,(LIS_QUAD *)tmpdot2.hi);
/* x = x + alpha*p + omega*ms */
/* r = s - omega*ams */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,p,x);
lis_vector_axpyex_mmm(omega,ms,x);
lis_quad_minus((LIS_QUAD *)omega.hi);
lis_vector_axpyzex_mmmm(omega,ams,s,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->residual[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) printf("iter: %5d residual = %e\n", iter, nrm2);
}
if( tol >= nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptimes = ptimes;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* z = M^-1 * r */
/* rho = <rtld,z> */
times = lis_wtime();
lis_psolve(solver, r, z);
ptimes += lis_wtime()-times;
lis_vector_dotex_mmm(rtld,z,&rho);
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) * (alpha / omega) */
/* p = z + beta*(p - omega*map) */
lis_quad_minus((LIS_QUAD *)omega.hi);
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_quad_div((LIS_QUAD *)tmpdot1.hi,(LIS_QUAD *)alpha.hi,(LIS_QUAD *)omega.hi);
lis_quad_mul((LIS_QUAD *)beta.hi,(LIS_QUAD *)beta.hi,(LIS_QUAD *)tmpdot1.hi);
lis_quad_minus((LIS_QUAD *)omega.hi);
lis_vector_axpyex_mmm(omega,map,p);
lis_vector_xpayex_mmm(z,beta,p);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicgsafe_switch(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_VECTOR x;
LIS_VECTOR r, rtld, rhat, p, ptld, phat;
LIS_VECTOR t, ttld, that, t0, t0hat;
LIS_VECTOR y, w, u, z;
LIS_QUAD_PTR alpha, beta, rho, rho_old;
LIS_QUAD_PTR qsi, eta, one;
LIS_QUAD_PTR tmp, tmpdot[5];
LIS_REAL bnrm2, nrm2, tol, tol2;
LIS_INT iter,maxiter,output,conv;
LIS_INT iter2,maxiter2;
double time,ptime;
LIS_DEBUG_FUNC_IN;
A = solver->A;
x = solver->x;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
maxiter2 = solver->options[LIS_OPTIONS_SWITCH_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
tol = solver->params[LIS_PARAMS_RESID-LIS_OPTIONS_LEN];
tol2 = solver->params[LIS_PARAMS_SWITCH_RESID-LIS_OPTIONS_LEN];
ptime = 0.0;
rtld = solver->work[0];
r = solver->work[1];
rhat = solver->work[2];
p = solver->work[3];
ptld = solver->work[4];
phat = solver->work[5];
t = solver->work[6];
ttld = solver->work[7];
that = solver->work[8];
t0 = solver->work[9];
t0hat = solver->work[10];
y = solver->work[11];
w = solver->work[12];
u = solver->work[13];
z = solver->work[14];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(qsi,4,1);
LIS_QUAD_SCALAR_MALLOC(eta,5,1);
LIS_QUAD_SCALAR_MALLOC(tmp,6,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[0],7,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[1],8,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[2],9,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[3],10,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[4],11,1);
LIS_QUAD_SCALAR_MALLOC(one,13,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
alpha.hi[0] = 1.0;
alpha.lo[0] = 0.0;
qsi.hi[0] = 1.0;
qsi.lo[0] = 0.0;
one.hi[0] = -1.0;
one.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol2 = solver->tol_switch;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, ttld);
lis_vector_set_allex_nm(0.0, ptld);
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, u);
lis_vector_set_allex_nm(0.0, t);
lis_vector_set_allex_nm(0.0, t0);
for( iter=1; iter<=maxiter2; iter++ )
{
/* rho = <rtld,r> */
lis_vector_dot(rtld,r,&rho.hi[0]);
/* test breakdown */
if( rho.hi[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) * (alpha / qsi) */
beta.hi[0] = (rho.hi[0] / rho_old.hi[0]) * (alpha.hi[0] / qsi.hi[0]);
/* w = ttld + beta*ptld */
lis_vector_axpyz(beta.hi[0],ptld,ttld,w);
/* rhat = M^-1 * r */
time = lis_wtime();
lis_psolve(solver, r, rhat);
ptime += lis_wtime()-time;
/* p = rhat + beta*(p - u) */
lis_vector_axpy(-1,u,p);
lis_vector_xpay(rhat,beta.hi[0],p);
/* ptld = A * p */
lis_matvec(A,p,ptld);
/* tmpdot[0] = <rtld,ptld> */
lis_vector_dot(rtld,ptld,&tmpdot[0].hi[0]);
/* test breakdown */
/* */
/* alpha = rho / tmpdot[0] */
alpha.hi[0] = rho.hi[0] / tmpdot[0].hi[0];
/* y = t - r + alpha*(-w + ptld) */
lis_vector_axpyz(-1,w,ptld,y);
lis_vector_xpay(t,alpha.hi[0],y);
lis_vector_axpy(-1,r,y);
/* t = r - alpha*ptld */
lis_vector_axpyz(-alpha.hi[0],ptld,r,t);
/* that = M^-1 * t */
/* phat = M^-1 * ptld */
/* t0hat = M^-1 * t0 */
time = lis_wtime();
lis_psolve(solver, t, that);
lis_psolve(solver, ptld, phat);
lis_psolve(solver, t0, t0hat);
ptime += lis_wtime()-time;
/* ttld = A * that */
lis_matvec(A,that,ttld);
/* tmpdot[0] = <y,y> */
/* tmpdot[1] = <ttld,t> */
/* tmpdot[2] = <y,t> */
/* tmpdot[3] = <ttld,y> */
/* tmpdot[4] = <ttld,ttld> */
lis_vector_dot(y,y,&tmpdot[0].hi[0]);
lis_vector_dot(ttld,t,&tmpdot[1].hi[0]);
lis_vector_dot(y,t,&tmpdot[2].hi[0]);
lis_vector_dot(ttld,y,&tmpdot[3].hi[0]);
lis_vector_dot(ttld,ttld,&tmpdot[4].hi[0]);
if(iter==1)
{
qsi.hi[0] = tmpdot[1].hi[0] / tmpdot[4].hi[0];
eta.hi[0] = 0.0;
}
else
{
tmp.hi[0] = tmpdot[4].hi[0]*tmpdot[0].hi[0] - tmpdot[3].hi[0]*tmpdot[3].hi[0];
qsi.hi[0] = (tmpdot[0].hi[0]*tmpdot[1].hi[0] - tmpdot[2].hi[0]*tmpdot[3].hi[0]) / tmp.hi[0];
eta.hi[0] = (tmpdot[4].hi[0]*tmpdot[2].hi[0] - tmpdot[3].hi[0]*tmpdot[1].hi[0]) / tmp.hi[0];
}
/* u = qsi*phat + eta*(t0hat - rhat + beta*u) */
lis_vector_xpay(t0hat,beta.hi[0],u);
lis_vector_axpy(-1,rhat,u);
lis_vector_scale(eta.hi[0],u);
lis_vector_axpy(qsi.hi[0],phat,u);
/* z = qsi*rhat + eta*z - alpha*u */
lis_vector_scale(eta.hi[0],z);
lis_vector_axpy(qsi.hi[0],rhat,z);
lis_vector_axpy(-alpha.hi[0],u,z);
/* x = x + alpha*p + z */
lis_vector_axpy(alpha.hi[0],p,x);
lis_vector_axpy(1,z,x);
/* r = t - eta*y - qsi*ttld */
lis_vector_axpyz(-eta.hi[0],y,t,r);
lis_vector_axpy(-qsi.hi[0],ttld,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol2 >= nrm2 )
{
solver->iter = iter;
solver->iter2 = iter;
solver->ptime = ptime;
break;
}
lis_vector_copy(t,t0);
rho_old.hi[0] = rho.hi[0];
}
r->precision = LIS_PRECISION_QUAD;
p->precision = LIS_PRECISION_QUAD;
t->precision = LIS_PRECISION_QUAD;
t0->precision = LIS_PRECISION_QUAD;
ptld->precision = LIS_PRECISION_QUAD;
that->precision = LIS_PRECISION_QUAD;
solver->options[LIS_OPTIONS_INITGUESS_ZEROS] = LIS_FALSE;
lis_vector_copyex_mn(x,solver->xx);
rho_old.hi[0] = 1.0;
alpha.hi[0] = 1.0;
qsi.hi[0] = 1.0;
one.hi[0] = -1.0;
/* Initial Residual */
lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2);
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, ttld);
lis_vector_set_allex_nm(0.0, ptld);
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, u);
lis_vector_set_allex_nm(0.0, t);
lis_vector_set_allex_nm(0.0, t0);
for( iter2=iter+1; iter2<=maxiter; iter2++ )
{
/* rho = <rtld,r> */
lis_vector_dotex_mmm(rtld,r,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) * (alpha / qsi) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_quad_div((LIS_QUAD *)tmp.hi,(LIS_QUAD *)alpha.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)beta.hi,(LIS_QUAD *)beta.hi,(LIS_QUAD *)tmp.hi);
/* w = ttld + beta*ptld */
lis_vector_axpyzex_mmmm(beta,ptld,ttld,w);
/* rhat = M^-1 * r */
time = lis_wtime();
lis_psolve(solver, r, rhat);
ptime += lis_wtime()-time;
/* p = rhat + beta*(p - u) */
lis_vector_axpyex_mmm(one,u,p);
lis_vector_xpayex_mmm(rhat,beta,p);
/* ptld = A * p */
lis_matvec(A,p,ptld);
/* tmpdot[0] = <rtld,ptld> */
lis_vector_dotex_mmm(rtld,ptld,&tmpdot[0]);
/* test breakdown */
/* */
/* alpha = rho / tmpdot[0] */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot[0].hi);
/* y = t - r + alpha*(-w + ptld) */
lis_vector_axpyzex_mmmm(one,w,ptld,y);
lis_vector_xpayex_mmm(t,alpha,y);
lis_vector_axpyex_mmm(one,r,y);
/* t = r - alpha*ptld */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyzex_mmmm(alpha,ptld,r,t);
/* that = M^-1 * t */
/* phat = M^-1 * ptld */
/* t0hat = M^-1 * t0 */
time = lis_wtime();
lis_psolve(solver, t, that);
lis_psolve(solver, ptld, phat);
lis_psolve(solver, t0, t0hat);
ptime += lis_wtime()-time;
/* ttld = A * that */
lis_matvec(A,that,ttld);
/* tmpdot[0] = <y,y> */
/* tmpdot[1] = <ttld,t> */
/* tmpdot[2] = <y,t> */
/* tmpdot[3] = <ttld,y> */
/* tmpdot[4] = <ttld,ttld> */
lis_vector_dotex_mmm(y,y,&tmpdot[0]);
lis_vector_dotex_mmm(ttld,t,&tmpdot[1]);
lis_vector_dotex_mmm(y,t,&tmpdot[2]);
lis_vector_dotex_mmm(ttld,y,&tmpdot[3]);
lis_vector_dotex_mmm(ttld,ttld,&tmpdot[4]);
if(iter==1)
{
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[1].hi,(LIS_QUAD *)tmpdot[4].hi);
eta.hi[0] = 0.0;
eta.lo[0] = 0.0;
}
else
{
lis_quad_mul((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_sqr((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmp.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[2].hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)eta.hi);
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmp.hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[2].hi);
lis_quad_mul((LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[3].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_sub((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_div((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmp.hi);
}
/* u = qsi*phat + eta*(t0hat - rhat + beta*u) */
lis_vector_xpayex_mmm(t0hat,beta,u);
lis_vector_axpyex_mmm(one,rhat,u);
lis_vector_scaleex_mm(eta,u);
lis_vector_axpyex_mmm(qsi,phat,u);
/* z = qsi*rhat + eta*z - alpha*u */
lis_vector_scaleex_mm(eta,z);
lis_vector_axpyex_mmm(qsi,rhat,z);
lis_vector_axpyex_mmm(alpha,u,z);
/* x = x + alpha*p + z */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_quad_minus((LIS_QUAD *)one.hi);
lis_vector_axpyex_mmm(alpha,p,x);
lis_vector_axpyex_mmm(one,z,x);
lis_quad_minus((LIS_QUAD *)one.hi);
/* r = t - eta*y - qsi*ttld */
lis_quad_minus((LIS_QUAD *)eta.hi);
lis_quad_minus((LIS_QUAD *)qsi.hi);
lis_vector_axpyzex_mmmm(eta,y,t,r);
lis_vector_axpyex_mmm(qsi,ttld,r);
lis_quad_minus((LIS_QUAD *)eta.hi);
lis_quad_minus((LIS_QUAD *)qsi.hi);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter2] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter2;
solver->iter2 = iter;
solver->resid = nrm2;
solver->ptime = ptime;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
lis_vector_copyex_mm(t,t0);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->iter2 = iter2;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicgsafe_quad(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_VECTOR x;
LIS_VECTOR r, rtld, rhat, p, ptld;
LIS_VECTOR t, ttld;
LIS_VECTOR y, v, u, utld, z;
LIS_QUAD_PTR alpha, beta, rho, rho_old;
LIS_QUAD_PTR qsi, eta;
LIS_QUAD_PTR tmp, tmpdot[5],one;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,output,conv;
double time,ptime;
LIS_DEBUG_FUNC_IN;
A = solver->A;
x = solver->x;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptime = 0.0;
rtld = solver->work[0];
r = solver->work[1];
rhat = solver->work[2];
p = solver->work[3];
ptld = solver->work[4];
t = solver->work[5];
ttld = solver->work[6];
y = solver->work[7];
v = solver->work[8];
u = solver->work[9];
z = solver->work[10];
utld = solver->work[11];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(qsi,4,1);
LIS_QUAD_SCALAR_MALLOC(eta,5,1);
LIS_QUAD_SCALAR_MALLOC(tmp,6,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[0],7,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[1],8,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[2],9,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[3],10,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[4],11,1);
LIS_QUAD_SCALAR_MALLOC(one,13,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
alpha.hi[0] = 1.0;
alpha.lo[0] = 0.0;
qsi.hi[0] = 1.0;
qsi.lo[0] = 0.0;
one.hi[0] = -1.0;
one.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0,p);
lis_vector_set_allex_nm(0.0,u);
lis_vector_set_allex_nm(0.0,ptld);
lis_vector_set_allex_nm(0.0,utld);
for( iter=1; iter<=maxiter; iter++ )
{
/* rho = <rtld,r> */
lis_vector_dotex_mmm(rtld,r,&rho);
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) * (alpha / qsi) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_quad_div((LIS_QUAD *)tmp.hi,(LIS_QUAD *)alpha.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)beta.hi,(LIS_QUAD *)beta.hi,(LIS_QUAD *)tmp.hi);
/* rhat = M^-1 * r */
/* v = A * rhat */
time = lis_wtime();
lis_psolve(solver, r, rhat);
ptime += lis_wtime()-time;
lis_matvec(A,rhat,v);
/* p = rhat + beta*(p - u) */
lis_vector_axpyex_mmm(one,u,p);
lis_vector_xpayex_mmm(rhat,beta,p);
/* ptld = v + beta*(ptld - utld) */
lis_vector_axpyex_mmm(one,utld,ptld);
lis_vector_xpayex_mmm(v,beta,ptld);
/* tmpdot[0] = <rtld,ptld> */
lis_vector_dotex_mmm(rtld,ptld,&tmpdot[0]);
/* test breakdown */
/* */
/* alpha = rho / tmpdot[0] */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot[0].hi);
/* tmpdot[0] = <y,y> */
/* tmpdot[1] = <v,r> */
/* tmpdot[2] = <y,r> */
/* tmpdot[3] = <v,y> */
/* tmpdot[4] = <v,v> */
lis_vector_dotex_mmm(y,y,&tmpdot[0]);
lis_vector_dotex_mmm(v,r,&tmpdot[1]);
lis_vector_dotex_mmm(y,r,&tmpdot[2]);
lis_vector_dotex_mmm(v,y,&tmpdot[3]);
lis_vector_dotex_mmm(v,v,&tmpdot[4]);
if(iter==1)
{
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[1].hi,(LIS_QUAD *)tmpdot[4].hi);
eta.hi[0] = 0.0;
eta.lo[0] = 0.0;
}
else
{
lis_quad_mul((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_sqr((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmp.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[2].hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)eta.hi);
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmp.hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[2].hi);
lis_quad_mul((LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[3].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_sub((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_div((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmp.hi);
}
/* t = qsi*ptld + eta*y */
lis_vector_copyex_mm(y,t);
lis_vector_scaleex_mm(eta,t);
lis_vector_axpyex_mmm(qsi,ptld,t);
/* ttld = M^-1 * t */
time = lis_wtime();
lis_psolve(solver, t, ttld);
ptime += lis_wtime()-time;
/* u = ttld + eta*beta*u */
/* utld = A * u */
lis_quad_mul((LIS_QUAD *)tmp.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)beta.hi);
lis_vector_xpayex_mmm(ttld,tmp,u);
lis_matvec(A,u,utld);
/* z = qsi*rhat + eta*z - alpha*u */
lis_vector_scaleex_mm(eta,z);
lis_vector_axpyex_mmm(qsi,rhat,z);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,u,z);
/* y = qsi*v + eta*y - alpha*utld */
lis_vector_scaleex_mm(eta,y);
lis_vector_axpyex_mmm(qsi,v,y);
lis_vector_axpyex_mmm(alpha,utld,y);
lis_quad_minus((LIS_QUAD *)alpha.hi);
/* x = x + alpha*p + z */
lis_vector_axpyex_mmm(alpha,p,x);
lis_quad_minus((LIS_QUAD *)one.hi);
lis_vector_axpyex_mmm(one,z,x);
lis_quad_minus((LIS_QUAD *)one.hi);
/* r = r - alpha*ptld - y */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,ptld,r);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(one,y,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptime = ptime;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicrsafe_quad(LIS_SOLVER solver)
{
LIS_MATRIX A;
LIS_VECTOR x;
LIS_VECTOR r, rtld, artld, mr, amr, p, ap, map;
LIS_VECTOR y, my, u, au, z;
LIS_QUAD_PTR alpha, beta, rho, rho_old;
LIS_QUAD_PTR qsi, eta, one;
LIS_QUAD_PTR tmp, tmpdot[5];
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,output,conv;
double time,ptime;
LIS_DEBUG_FUNC_IN;
A = solver->A;
x = solver->x;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptime = 0.0;
rtld = solver->work[0];
r = solver->work[1];
mr = solver->work[2];
amr = solver->work[3];
p = solver->work[4];
ap = solver->work[5];
map = solver->work[6];
my = solver->work[7];
y = solver->work[8];
u = solver->work[9];
z = solver->work[10];
au = solver->work[11];
artld = solver->work[12];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(qsi,4,1);
LIS_QUAD_SCALAR_MALLOC(eta,5,1);
LIS_QUAD_SCALAR_MALLOC(tmp,6,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[0],7,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[1],8,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[2],9,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[3],10,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot[4],11,1);
LIS_QUAD_SCALAR_MALLOC(one,13,1);
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_matvect(A,rtld,artld);
time = lis_wtime();
lis_psolve(solver, r, mr);
ptime += lis_wtime()-time;
lis_matvec(A,mr,amr);
lis_vector_dotex_mmm(rtld,amr,&rho_old);
lis_vector_copyex_mm(amr,ap);
lis_vector_copyex_mm(mr,p);
one.hi[0] = -1.0;
one.lo[0] = 0.0;
for( iter=1; iter<=maxiter; iter++ )
{
/* map = M^-1 * ap */
time = lis_wtime();
lis_psolve(solver, ap, map);
ptime += lis_wtime()-time;
/* tmpdot[0] = <artld,map> */
/* alpha = rho_old / tmpdot[0] */
lis_vector_dotex_mmm(artld,map,&tmpdot[0]);
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho_old.hi,(LIS_QUAD *)tmpdot[0].hi);
/* tmpdot[0] = <y,y> */
/* tmpdot[1] = <amr,r> */
/* tmpdot[2] = <y,r> */
/* tmpdot[3] = <amr,y> */
/* tmpdot[4] = <amr,amr> */
lis_vector_dotex_mmm(y,y,&tmpdot[0]);
lis_vector_dotex_mmm(amr,r,&tmpdot[1]);
lis_vector_dotex_mmm(y,r,&tmpdot[2]);
lis_vector_dotex_mmm(amr,y,&tmpdot[3]);
lis_vector_dotex_mmm(amr,amr,&tmpdot[4]);
if(iter==1)
{
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[1].hi,(LIS_QUAD *)tmpdot[4].hi);
eta.hi[0] = 0.0;
eta.lo[0] = 0.0;
}
else
{
lis_quad_mul((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_sqr((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)tmp.hi,(LIS_QUAD *)tmp.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[2].hi,(LIS_QUAD *)tmpdot[3].hi);
lis_quad_sub((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)eta.hi);
lis_quad_div((LIS_QUAD *)qsi.hi,(LIS_QUAD *)qsi.hi,(LIS_QUAD *)tmp.hi);
lis_quad_mul((LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[4].hi,(LIS_QUAD *)tmpdot[2].hi);
lis_quad_mul((LIS_QUAD *)tmpdot[0].hi,(LIS_QUAD *)tmpdot[3].hi,(LIS_QUAD *)tmpdot[1].hi);
lis_quad_sub((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmpdot[0].hi);
lis_quad_div((LIS_QUAD *)eta.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)tmp.hi);
}
/* u = qsi*map + eta*my + eta*beta*u */
/* au = A * u */
lis_quad_mul((LIS_QUAD *)tmp.hi,(LIS_QUAD *)eta.hi,(LIS_QUAD *)beta.hi);
lis_vector_scaleex_mm(tmp,u);
lis_vector_axpyex_mmm(qsi,map,u);
lis_vector_axpyex_mmm(eta,my,u);
lis_matvec(A,u,au);
/* z = qsi*mr + eta*z - alpha*u */
lis_vector_scaleex_mm(eta,z);
lis_vector_axpyex_mmm(qsi,mr,z);
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,u,z);
/* y = qsi*amr + eta*y - alpha*au */
/* my = M^-1 * y */
lis_vector_scaleex_mm(eta,y);
lis_vector_axpyex_mmm(qsi,amr,y);
lis_vector_axpyex_mmm(alpha,au,y);
time = lis_wtime();
lis_psolve(solver, y, my);
ptime += lis_wtime()-time;
/* x = x + alpha*p + z */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,p,x);
lis_quad_minus((LIS_QUAD *)one.hi);
lis_vector_axpyex_mmm(one,z,x);
/* r = r - alpha*ap - y */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_quad_minus((LIS_QUAD *)one.hi);
lis_vector_axpyex_mmm(alpha,ap,r);
lis_vector_axpyex_mmm(one,y,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol >= nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptime = ptime;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* mr = mr - alpha*map - my */
/* amr = A * mr */
/* rho = <rtld,amr> */
lis_vector_axpyex_mmm(alpha,map,mr);
lis_vector_axpyex_mmm(one,my,mr);
lis_matvec(A,mr,amr);
lis_vector_dotex_mmm(rtld,amr,&rho);
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) * (alpha / qsi) */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
lis_quad_div((LIS_QUAD *)tmp.hi,(LIS_QUAD *)alpha.hi,(LIS_QUAD *)qsi.hi);
lis_quad_mul((LIS_QUAD *)beta.hi,(LIS_QUAD *)beta.hi,(LIS_QUAD *)tmp.hi);
/* p = mr + beta*(p - u) */
/* ap = amr + beta*(ap - au) */
lis_vector_axpyex_mmm(one,u,p);
lis_vector_xpayex_mmm(mr,beta,p);
lis_vector_axpyex_mmm(one,au,ap);
lis_vector_xpayex_mmm(amr,beta,ap);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
LIS_INT lis_bicg_quad(LIS_SOLVER solver)
{
LIS_MATRIX A,At;
LIS_VECTOR x;
LIS_VECTOR r,rtld, z,ztld,p, ptld, q, qtld;
LIS_QUAD_PTR alpha, beta, rho, rho_old, tmpdot1;
LIS_REAL bnrm2, nrm2, tol;
LIS_INT iter,maxiter,output,conv;
double time,ptime;
LIS_DEBUG_FUNC_IN;
A = solver->A;
At = solver->A;
x = solver->x;
maxiter = solver->options[LIS_OPTIONS_MAXITER];
output = solver->options[LIS_OPTIONS_OUTPUT];
conv = solver->options[LIS_OPTIONS_CONV_COND];
ptime = 0.0;
r = solver->work[0];
rtld = solver->work[1];
z = solver->work[2];
ztld = solver->work[3];
p = solver->work[4];
ptld = solver->work[5];
q = solver->work[2];
qtld = solver->work[3];
LIS_QUAD_SCALAR_MALLOC(alpha,0,1);
LIS_QUAD_SCALAR_MALLOC(beta,1,1);
LIS_QUAD_SCALAR_MALLOC(rho,2,1);
LIS_QUAD_SCALAR_MALLOC(rho_old,3,1);
LIS_QUAD_SCALAR_MALLOC(tmpdot1,4,1);
rho_old.hi[0] = 1.0;
rho_old.lo[0] = 0.0;
/* Initial Residual */
if( lis_solver_get_initial_residual(solver,NULL,NULL,r,&bnrm2) )
{
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
tol = solver->tol;
lis_solver_set_shadowresidual(solver,r,rtld);
lis_vector_set_allex_nm(0.0, p);
lis_vector_set_allex_nm(0.0, ptld);
for( iter=1; iter<=maxiter; iter++ )
{
/* z = M^-1 * r */
/* ztld = M^-T * rtld */
time = lis_wtime();
lis_psolve(solver, r, z);
lis_psolvet(solver, rtld, ztld);
ptime += lis_wtime()-time;
/* rho = <z,rtld> */
lis_vector_dotex_mmm(z,rtld,&rho);
/* printf("rho = %e %e\n",rho.hi[0],rho.lo[0]);*/
/* test breakdown */
if( rho.hi[0]==0.0 && rho.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* beta = (rho / rho_old) */
lis_quad_div((LIS_QUAD *)beta.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)rho_old.hi);
/* printf("beta = %e %e\n",beta.hi[0],beta.lo[0]);*/
/* p = z + beta*p */
/* ptld = ztld + beta*ptld */
/* q = A * p */
/* qtld = A^T * ptld */
lis_vector_xpayex_mmm(z,beta,p);
lis_matvec(A,p,q);
lis_vector_xpayex_mmm(ztld,beta,ptld);
lis_matvect(At,ptld,qtld);
/* tmpdot1 = <ptld,q> */
lis_vector_dotex_mmm(ptld,q,&tmpdot1);
/* printf("tmpdot1 = %e %e\n",tmpdot1.hi[0],tmpdot1.lo[0]);*/
/* test breakdown */
if( tmpdot1.hi[0]==0.0 && tmpdot1.lo[0]==0.0 )
{
solver->retcode = LIS_BREAKDOWN;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_BREAKDOWN;
}
/* alpha = rho / tmpdot1 */
lis_quad_div((LIS_QUAD *)alpha.hi,(LIS_QUAD *)rho.hi,(LIS_QUAD *)tmpdot1.hi);
/* printf("alpha = %e %e\n",alpha.hi[0],alpha.lo[0]);*/
/* x = x + alpha*p */
lis_vector_axpyex_mmm(alpha,p,x);
/* r = r - alpha*q */
lis_quad_minus((LIS_QUAD *)alpha.hi);
lis_vector_axpyex_mmm(alpha,q,r);
/* convergence check */
lis_solver_get_residual[conv](r,solver,&nrm2);
if( output )
{
if( output & LIS_PRINT_MEM ) solver->rhistory[iter] = nrm2;
if( output & LIS_PRINT_OUT && A->my_rank==0 ) lis_print_rhistory(iter,nrm2);
}
if( tol > nrm2 )
{
solver->retcode = LIS_SUCCESS;
solver->iter = iter;
solver->resid = nrm2;
solver->ptime = ptime;
LIS_DEBUG_FUNC_OUT;
return LIS_SUCCESS;
}
/* rtld = rtld - alpha*qtld */
lis_vector_axpyex_mmm(alpha,qtld,rtld);
rho_old.hi[0] = rho.hi[0];
rho_old.lo[0] = rho.lo[0];
}
solver->retcode = LIS_MAXITER;
solver->iter = iter;
solver->resid = nrm2;
LIS_DEBUG_FUNC_OUT;
return LIS_MAXITER;
}
