def test_chained_full_resample_matrix(ctx_factory, ndim, visualize=False):
from meshmode.discretization.connection.chained import \
make_full_resample_matrix
ctx = ctx_factory()
queue = cl.CommandQueue(ctx)
discr = create_discretization(queue, ndim)
connections = []
conn = create_refined_connection(queue, discr)
connections.append(conn)
conn = create_refined_connection(queue, conn.to_discr)
connections.append(conn)
from meshmode.discretization.connection import \
ChainedDiscretizationConnection
chained = ChainedDiscretizationConnection(connections)
def f(x):
from six.moves import reduce
return 0.1 * reduce(lambda x, y: x * cl.clmath.sin(5 * y), x)
resample_mat = make_full_resample_matrix(queue, chained).get(queue)
x = connections[0].from_discr.nodes().with_queue(queue)
fx = f(x)
f1 = np.dot(resample_mat, fx.get(queue))
f2 = chained(queue, fx).get(queue)
f3 = connections[1](queue, connections[0](queue, fx)).get(queue)
assert np.allclose(f1, f2)
assert np.allclose(f2, f3)
def test_chained_full_resample_matrix(actx_factory, ndim, visualize=False):
from meshmode.discretization.connection.chained import \
make_full_resample_matrix
actx = actx_factory()
discr = create_discretization(actx, ndim, order=2, nelements=12)
connections = []
conn = create_refined_connection(actx, discr)
connections.append(conn)
conn = create_refined_connection(actx, conn.to_discr)
connections.append(conn)
from meshmode.discretization.connection import \
ChainedDiscretizationConnection
chained = ChainedDiscretizationConnection(connections)
def f(x):
from functools import reduce
return 0.1 * reduce(lambda x, y: x * actx.np.sin(5 * y), x)
resample_mat = actx.to_numpy(make_full_resample_matrix(actx, chained))
x = thaw(actx, connections[0].from_discr.nodes())
fx = f(x)
f1 = resample_mat @ actx.to_numpy(flatten(fx))
f2 = actx.to_numpy(flatten(chained(fx)))
f3 = actx.to_numpy(flatten(connections[1](connections[0](fx))))
assert np.allclose(f1, f2)
assert np.allclose(f2, f3)
def test_chained_full_resample_matrix(ctx_factory, ndim, visualize=False):
from meshmode.discretization.connection.chained import \
make_full_resample_matrix
ctx = ctx_factory()
queue = cl.CommandQueue(ctx)
discr = create_discretization(queue, ndim)
connections = []
conn = create_refined_connection(queue, discr)
connections.append(conn)
conn = create_refined_connection(queue, conn.to_discr)
connections.append(conn)
from meshmode.discretization.connection import \
ChainedDiscretizationConnection
chained = ChainedDiscretizationConnection(connections)
def f(x):
from six.moves import reduce
return 0.1 * reduce(lambda x, y: x * cl.clmath.sin(5 * y), x)
resample_mat = make_full_resample_matrix(queue, chained).get(queue)
x = connections[0].from_discr.nodes().with_queue(queue)
fx = f(x)
f1 = np.dot(resample_mat, fx.get(queue))
f2 = chained(queue, fx).get(queue)
f3 = connections[1](queue, connections[0](queue, fx)).get(queue)
assert np.allclose(f1, f2)
assert np.allclose(f2, f3)
def test_chained_full_resample_matrix(ctx_factory, ndim, visualize=False):
from meshmode.discretization.connection.chained import \
make_full_resample_matrix
ctx = ctx_factory()
queue = cl.CommandQueue(ctx)
actx = PyOpenCLArrayContext(queue)
discr = create_discretization(actx, ndim)
connections = []
conn = create_refined_connection(actx, discr)
connections.append(conn)
conn = create_refined_connection(actx, conn.to_discr)
connections.append(conn)
from meshmode.discretization.connection import \
ChainedDiscretizationConnection
chained = ChainedDiscretizationConnection(connections)
def f(x):
from six.moves import reduce
return 0.1 * reduce(lambda x, y: x * actx.np.sin(5 * y), x)
resample_mat = actx.to_numpy(make_full_resample_matrix(actx, chained))
x = thaw(actx, connections[0].from_discr.nodes())
fx = f(x)
f1 = resample_mat @ actx.to_numpy(flatten(fx))
f2 = actx.to_numpy(flatten(chained(fx)))
f3 = actx.to_numpy(flatten(connections[1](connections[0](fx))))
assert np.allclose(f1, f2)
assert np.allclose(f2, f3)