def setUp(self):
""" The setup consists of four particles placed at the
vertices of a unit square. The force function to be tested is:
..math::
f_i = \sum_{j=1}^{4} \frac{m_j}{|x_j - x_i|^3 +
\eps}(x_j - x_i)
The mass of each particle is 1
"""
self.precision = "single"
self.np = 4
x = numpy.array([0, 0, 1, 1], numpy.float64)
y = numpy.array([0, 1, 1, 0], numpy.float64)
z = numpy.zeros_like(x)
m = numpy.ones_like(x)
tmpx = numpy.zeros_like(x)
tmpy = numpy.zeros_like(x)
tmpz = numpy.zeros_like(x)
self.pa = pa = base.get_particle_array(name="test",
x=x,
y=y,
z=z,
m=m,
tmpx=tmpx,
tmpy=tmpy,
tmpz=tmpz,
cl_precision=self.precision)
self.func = func = sph.NBodyForce.get_func(pa, pa)
self.eps = func.eps
if solver.HAS_CL:
self.ctx = ctx = cl.create_some_context()
self.q = q = cl.CommandQueue(ctx)
pa.setup_cl(ctx, q)
pysph_root = solver.get_pysph_root()
template = solver.cl_read(
path.join(pysph_root, "sph/funcs/external_force.clt"),
function_name=func.cl_kernel_function_name,
precision=self.precision)
prog_src = solver.create_program(template, func)
self.prog = cl.Program(ctx,
prog_src).build(solver.get_cl_include())
def setUp(self):
""" The setup consists of four particles placed at the
vertices of a unit square. The force function to be tested is:
..math::
f_i = \sum_{j=1}^{4} \frac{m_j}{|x_j - x_i|^3 +
\eps}(x_j - x_i)
The mass of each particle is 1
"""
self.precision = "single"
self.np = 4
x = numpy.array([0, 0, 1, 1], numpy.float64)
y = numpy.array([0, 1, 1, 0], numpy.float64)
z = numpy.zeros_like(x)
m = numpy.ones_like(x)
tmpx = numpy.zeros_like(x)
tmpy = numpy.zeros_like(x)
tmpz = numpy.zeros_like(x)
self.pa = pa = base.get_particle_array(name="test", x=x, y=y, z=z,
m=m, tmpx=tmpx, tmpy=tmpy,
tmpz=tmpz,
cl_precision=self.precision)
self.func = func = sph.NBodyForce.get_func(pa, pa)
self.eps = func.eps
if solver.HAS_CL:
self.ctx = ctx = cl.create_some_context()
self.q = q = cl.CommandQueue(ctx)
pa.setup_cl(ctx, q)
pysph_root = solver.get_pysph_root()
template = solver.cl_read(
path.join(pysph_root, "sph/funcs/external_force.clt"),
function_name=func.cl_kernel_function_name,
precision=self.precision)
prog_src = solver.create_program(template, func)
self.prog = cl.Program(ctx, prog_src).build(solver.get_cl_include())
def setup_cl(self):
pa = self.pa
if solver.HAS_CL:
self.ctx = ctx = cl.create_some_context()
self.q = q = cl.CommandQueue(ctx)
pa.setup_cl(ctx, q)
pysph_root = solver.get_pysph_root()
template = solver.cl_read(
path.join(pysph_root, "sph/funcs/pressure_funcs.clt"),
function_name=self.grad_func.cl_kernel_function_name,
precision=self.precision)
prog_src = solver.create_program(template, self.grad_func)
self.prog=cl.Program(ctx, prog_src).build(solver.get_cl_include())