def test2d(self):
n = 8
# global number of cells
ns = (n, n)
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print(
'*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs'
)
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
nsLocal = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
nsLocal.append(iend - ibeg)
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h * (numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
op = StencilOperator(dc, periodic=(False, False))
for i in range(ndims):
disp = [0] * ndims
for pm in (-1, 1):
disp[i] = pm
op.addStencilBranch(tuple(disp), 1.0)
op.addStencilBranch(tuple([0] * ndims), -2 * ndims)
# set the input function
xx = numpy.outer(axes[0], numpy.ones((nsLocal[1], )))
yy = numpy.outer(numpy.ones((nsLocal[0], )), axes[1])
inp = 0.5 * xx * yy**2
#print('[{0}] inp = {1}'.format(self.rk, str(inp)))
out = op.apply(
inp) / hs[0]**2 # NEED TO ADJUST WHEN CELL SIZE IS DIFFERENT IN Y!
#print('[{0}] out = {1}'.format(self.rk, str(out)))
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test2d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -198.0), 1.e-10)
def test2d(self):
n = 8
# global number of cells
ns = (n, n)
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print('*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs')
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
nsLocal = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
nsLocal.append(iend - ibeg)
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h*(numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
op = StencilOperator(dc, periodic=(False, False))
for i in range(ndims):
disp = [0] * ndims
for pm in (-1, 1):
disp[i] = pm
op.addStencilBranch(tuple(disp), 1.0)
op.addStencilBranch(tuple([0]*ndims), -2*ndims)
# set the input function
xx = numpy.outer(axes[0], numpy.ones((nsLocal[1],)))
yy = numpy.outer(numpy.ones((nsLocal[0],)), axes[1])
inp = 0.5 * xx * yy ** 2
#print('[{0}] inp = {1}'.format(self.rk, str(inp)))
out = op.apply(inp) / hs[0]**2 # NEED TO ADJUST WHEN CELL SIZE IS DIFFERENT IN Y!
#print('[{0}] out = {1}'.format(self.rk, str(out)))
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test2d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -198.0), 1.e-10)
def test1d(self):
n = 8
# global number of cells
ns = (n, )
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print(
'*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs'
)
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h * (numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
op = StencilOperator(dc, periodic=(False, ))
op.addStencilBranch((1, ), 1.0)
op.addStencilBranch((-1, ), 1.0)
op.addStencilBranch((0, ), -2.0)
# set the input function
inp = 0.5 * axes[0]**2
#print('[{0}] inp = {1}'.format(self.rk, str(inp)))
out = op.apply(inp) / hs[0]**2
#print('[{0}] out = {1}'.format(self.rk, str(out)))
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test1d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -28.25), 1.e-10)
def test1d(self):
n = 8
# global number of cells
ns = (n,)
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print('*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs')
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h*(numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
op = StencilOperator(dc, periodic=(False,))
op.addStencilBranch((1,), 1.0)
op.addStencilBranch((-1,), 1.0)
op.addStencilBranch((0,), -2.0)
# set the input function
inp = 0.5 * axes[0]**2
#print('[{0}] inp = {1}'.format(self.rk, str(inp)))
out = op.apply(inp) / hs[0]**2
#print('[{0}] out = {1}'.format(self.rk, str(out)))
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test1d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -28.25), 1.e-10)
def test3d(self):
n = 8
# global number of cells
ns = (n, n, n)
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print(
'*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs'
)
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
nsLocal = []
iBegs = []
iEnds = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
iBegs.append(ibeg)
iEnds.append(iend)
nsLocal.append(iend - ibeg)
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h * (numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
op = StencilOperator(dc, periodic=(False, False, True))
for i in range(ndims):
disp = [0] * ndims
for pm in (-1, 1):
disp[i] = pm
op.addStencilBranch(tuple(disp), 1.0)
op.addStencilBranch(tuple([0] * ndims), -2 * ndims)
# set the input function
inp = numpy.zeros(
(iEnds[0] - iBegs[0], iEnds[1] - iBegs[1], iEnds[2] - iBegs[2]),
numpy.float64)
for ig in range(iBegs[0], iEnds[0]):
i = ig - iBegs[0]
x = axes[0][i]
for jg in range(iBegs[1], iEnds[1]):
j = jg - iBegs[1]
y = axes[1][j]
for kg in range(iBegs[2], iEnds[2]):
k = kg - iBegs[2]
z = axes[2][k]
inp[i, j, k] = 0.5 * x * y**2
# check sum of input
localChkSum = numpy.sum(inp.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test3d check sum of input = {}'.format(chksum))
out = op.apply(inp)
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test3d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -24.75), 1.e-10)
def test3d(self):
n = 8
# global number of cells
ns = (n, n, n)
# domain decomposition
dc = CubeDecomp(self.sz, ns)
if not dc.getDecomp():
print('*** ERROR Invalid domain decomposition -- rerun with different sizes/number of procs')
sys.exit(1)
ndims = dc.getNumDims()
# local start/end grid indices
slab = dc.getSlab(self.rk)
# global domain boundaries
xmins = numpy.array([0.0 for i in range(ndims)])
xmaxs = numpy.array([1.0 for i in range(ndims)])
# local cell centered coordinates
axes = []
hs = []
nsLocal = []
iBegs = []
iEnds = []
for i in range(ndims):
ibeg, iend = slab[i].start, slab[i].stop
iBegs.append(ibeg)
iEnds.append(iend)
nsLocal.append(iend - ibeg)
h = (xmaxs[i] - xmins[i]) / float(ns[i])
ax = xmins[i] + h*(numpy.arange(ibeg, iend) + 0.5)
hs.append(h)
axes.append(ax)
lapl = Laplacian(dc, periodic=(False, False, True))
# set the input function
inp = numpy.zeros((iEnds[0] - iBegs[0], iEnds[1] - iBegs[1], iEnds[2] - iBegs[2]), numpy.float64)
for ig in range(iBegs[0], iEnds[0]):
i = ig - iBegs[0]
x = axes[0][i]
for jg in range(iBegs[1], iEnds[1]):
j = jg - iBegs[1]
y = axes[1][j]
for kg in range(iBegs[2], iEnds[2]):
k = kg - iBegs[2]
z = axes[2][k]
inp[i, j, k] = 0.5 * x * y**2
# check sum of input
localChkSum = numpy.sum(inp.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test3d check sum of input = {}'.format(chksum))
out = lapl.apply(inp)
# check sum
localChkSum = numpy.sum(out.flat)
chksum = numpy.sum(MPI.COMM_WORLD.gather(localChkSum, 0))
if self.rk == 0:
print('test3d check sum = {}'.format(chksum))
self.assertLessEqual(abs(chksum - -24.75), 1.e-10)