def test_shewchuck():
# Example from Shewchuck, converges in 2 iterations
A = Matrix([[3, 2], [2, 6]])
b = np.array([[2, 2, 1], [-8, 8, -4]], dtype=np.float64)
x, info = conjugate_gradients(A, b, maxit=200, eps=10**-8)
bhat = compute_array(A * x)
assert np.all(ndnorm(bhat - b, axis=0) / ndnorm(b, axis=0) < 1e-8)
assert info['iterations'] == 2
def test_shewchuck():
# Example from Shewchuck, converges in 2 iterations
A = Matrix([[3, 2], [2, 6]])
b = np.array([[2, 2, 1], [-8, 8, -4]], dtype=np.float64)
x, info = conjugate_gradients(A, b, maxit=200, eps=10 ** -8)
bhat = compute_array(A * x)
assert np.all(ndnorm(bhat - b, axis=0) / ndnorm(b, axis=0) < 1e-8)
assert info["iterations"] == 2
def conjugate_gradients(A, b, preconditioner=None, x0=None,
maxit=10**3, eps=10**-8,
relative_eps=True,
raise_error=True,
norm_order=None, logger=None, compiler=None,
stop_rule='residual'):
"""
"""
assert stop_rule in ('preconditioned_residual', 'residual')
b = np.asarray(b)
if x0 is None:
x0 = np.zeros(b.shape, dtype=b.dtype, order='F')
if preconditioner is None:
preconditioner = Matrix(np.ones(b.shape[0], dtype=b.dtype), diagonal=True)
info = {}
# Terminology/variable names follow Shewchuk, ch. B3
# r - residual
# d - preconditioned residual, "P r"
#
# P = inv(M)
r = b - compute_array(A * x0, compiler=compiler)
d = compute_array(preconditioner * r, compiler=compiler).copy('F')
delta_0 = delta_new = np.sum(r * d, axis=0)
if stop_rule == 'preconditioned_residual':
stop_treshold = eps**2 * delta_0
elif stop_rule == 'residual':
stop_treshold = eps**2 * np.dot(r.T, r)[0, 0]
info['residuals'] = residuals = [delta_0]
info['error'] = None
x = x0
for k in xrange(maxit):
if stop_rule == 'preconditioned_residual':
stop_measure = delta_new
elif stop_rule == 'residual':
stop_measure = np.dot(r.T, r)[0, 0]
logger.info('Iteration %d: %e (terminating at %e)', k,
np.sqrt(stop_measure), np.sqrt(stop_treshold))
if stop_measure < stop_treshold:
info['iterations'] = k
return (x, info)
q = compute_array(A * d, compiler=compiler)
dAd = np.sum(d * q)
if not np.isfinite(dAd):
raise AssertionError("conjugate_gradients: A * d yielded inf values")
if dAd == 0:
raise AssertionError("conjugate_gradients: A is singular")
alpha = delta_new / dAd
x += alpha * d
r -= alpha * q
if k > 0 and k % 50 == 0:
r_est = r
r = b - compute_array(A * x)
logger.info('Recomputing residual, relative error in estimate: %e',
np.linalg.norm(r - r_est) / np.linalg.norm(r))
s = compute_array(preconditioner * r, compiler=compiler)
delta_old = delta_new
delta_new = np.sum(r * s, axis=0)
beta = delta_new / delta_old
d = s + beta * d
residuals.append(delta_new)
err = ConvergenceError("Did not converge in %d iterations" % maxit)
if raise_error:
raise err
else:
info['iterations'] = maxit
info['error'] = err
return (x, info)
