def test_nontrivial_code(ctx):
argnames = (('output', ), ('input', ), ('param', ))
elw = Elementwise(ctx).set_argnames(*argnames)
code = dict(kernel="""
${input.ctype} a1 = ${input.load}(idx);
${input.ctype} a2 = ${input.load}(idx + ${size});
${output.store}(idx, a1 + test(a2, ${param}));
""",
functions="""
WITHIN_KERNEL ${output.ctype} test(${input.ctype} val, ${param.ctype} param)
{
return val + param;
}
""")
N = 1000
a = get_test_array(N * 2, numpy.float32)
a_dev = ctx.to_device(a)
b_dev = ctx.allocate(N, numpy.float32)
param = 1
elw.prepare_for(b_dev, a_dev, numpy.float32(param), code=code)
elw(b_dev, a_dev, param)
assert diff_is_negligible(ctx.from_device(b_dev), a[:N] + a[N:] + param)
def test_nontrivial_code(ctx):
argnames = (('output',), ('input',), ('param',))
elw = Elementwise(ctx).set_argnames(*argnames)
code = dict(
kernel="""
${input.ctype} a1 = ${input.load}(idx);
${input.ctype} a2 = ${input.load}(idx + ${size});
${output.store}(idx, a1 + test(a2, ${param}));
""",
functions="""
WITHIN_KERNEL ${output.ctype} test(${input.ctype} val, ${param.ctype} param)
{
return val + param;
}
""")
N = 1000
a = get_test_array(N * 2, numpy.float32)
a_dev = ctx.to_device(a)
b_dev = ctx.allocate(N, numpy.float32)
param = 1
elw.prepare_for(b_dev, a_dev, numpy.float32(param), code=code)
elw(b_dev, a_dev, param)
assert diff_is_negligible(ctx.from_device(b_dev), a[:N] + a[N:] + param)
def test_errors(ctx):
argnames = (('output',), ('input',), ('param',))
elw = Elementwise(ctx).set_argnames(*argnames)
code = dict(kernel="""
${input.ctype} a1 = ${input.load}(idx);
${input.ctype} a2 = ${input.load}(idx + ${size});
${output.store}(idx, a1 + a2 + ${param});
""")
argtypes = dict(output=numpy.float32, input=numpy.float32,
param=numpy.float32)
N = 1000
a = get_test_array(N * 2, numpy.float32)
a_dev = ctx.to_device(a)
b_dev = ctx.allocate(N, numpy.float32)
param = 1
elw.prepare_for(b_dev, a_dev, numpy.float32(param), code=code)
elw(b_dev, a_dev, param)
assert diff_is_negligible(ctx.from_device(b_dev), a[:N] + a[N:] + param)
def test_errors(ctx):
argnames = (('output', ), ('input', ), ('param', ))
elw = Elementwise(ctx).set_argnames(*argnames)
code = dict(kernel="""
${input.ctype} a1 = ${input.load}(idx);
${input.ctype} a2 = ${input.load}(idx + ${size});
${output.store}(idx, a1 + a2 + ${param});
""")
argtypes = dict(output=numpy.float32,
input=numpy.float32,
param=numpy.float32)
N = 1000
a = get_test_array(N * 2, numpy.float32)
a_dev = ctx.to_device(a)
b_dev = ctx.allocate(N, numpy.float32)
param = 1
elw.prepare_for(b_dev, a_dev, numpy.float32(param), code=code)
elw(b_dev, a_dev, param)
assert diff_is_negligible(ctx.from_device(b_dev), a[:N] + a[N:] + param)
