def test_looplevel():
x, y = hl.Var('x'), hl.Var('y')
target = hl.get_jit_target_from_environment()
buffer_input = hl.Buffer(hl.UInt(8), [4, 4])
buffer_input.fill(123)
func_input = hl.Func("func_input")
func_input[x, y] = x + y
simple_compute_at = hl.LoopLevel()
simple = simplestub.generate(target, buffer_input, func_input, 3.5,
compute_level=simple_compute_at)
computed_output = hl.Func('computed_output')
computed_output[x, y] = simple[x, y] + 3
simple_compute_at.set(hl.LoopLevel(computed_output, x))
_realize_and_check(computed_output, 3)
def test_looplevel():
x, y = hl.Var('x'), hl.Var('y')
target = hl.get_jit_target_from_environment()
buffer_input = hl.Buffer(hl.UInt(8), [4, 4])
buffer_input.fill(123)
func_input = hl.Func("func_input")
func_input[x, y] = x + y
simple_compute_at = hl.LoopLevel()
simple = simplestub.generate(target, buffer_input, func_input, 3.5,
compute_level=simple_compute_at)
computed_output = hl.Func('computed_output')
computed_output[x, y] = simple[x, y] + 3
simple_compute_at.set(hl.LoopLevel(computed_output, x))
_realize_and_check(computed_output, 3)
def test_simplestub():
x, y = hl.Var(), hl.Var()
target = hl.get_jit_target_from_environment()
b_in = hl.Buffer(hl.UInt(8), [2, 2])
b_in.fill(123)
f_in = hl.Func("f")
f_in[x, y] = x + y
# ----------- Inputs by-position
f = simplestub.generate(target, b_in, f_in, 3.5)
_realize_and_check(f)
# ----------- Inputs by-name
f = simplestub.generate(target,
buffer_input=b_in,
func_input=f_in,
float_arg=3.5)
_realize_and_check(f)
# ----------- Inputs w/ mixed by-position and by-name
f = simplestub.generate(target, b_in, f_in, float_arg=3.5)
_realize_and_check(f)
f = simplestub.generate(target, b_in, float_arg=3.5, func_input=f_in)
_realize_and_check(f)
# ----------- Above set again, w/ GeneratorParam mixed in
k = 42
f = simplestub.generate(target, b_in, f_in, 3.5, offset=k)
_realize_and_check(f, k)
f = simplestub.generate(target,
offset=k,
buffer_input=b_in,
func_input=f_in,
float_arg=3.5)
_realize_and_check(f, k)
f = simplestub.generate(target, b_in, f_in, offset=k, float_arg=3.5)
_realize_and_check(f, k)
f = simplestub.generate(target,
b_in,
float_arg=3.5,
offset=k,
func_input=f_in)
_realize_and_check(f, k)
# ----------- Test various failure modes
try:
# too many positional args
f = simplestub.generate(target, b_in, f_in, 3.5, 4)
except RuntimeError as e:
assert 'Expected at most 3 positional args, but saw 4.' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Inputs that can't be converted to what the receiver needs (positional)
f = simplestub.generate(target, 3.141592, "happy")
except RuntimeError as e:
assert 'Unable to cast Python instance' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Inputs that can't be converted to what the receiver needs (named)
f = simplestub.generate(target, b_in, f_in, float_arg="bogus")
except RuntimeError as e:
assert 'Unable to cast Python instance' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Missing required inputs
f = simplestub.generate(target, b_in, f_in)
except RuntimeError as e:
assert "Generator Input named 'float_arg' was not specified." in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Input specified by both pos and kwarg
f = simplestub.generate(target, b_in, f_in, 3.5, float_arg=4.5)
except RuntimeError as e:
assert "Generator Input named 'float_arg' was specified by both position and keyword." in str(
e)
else:
assert False, 'Did not see expected exception!'
try:
# Bad input name
f = simplestub.generate(target,
b_in,
float_arg=3.5,
offset=k,
funk_input=f_in)
except RuntimeError as e:
assert "Generator Input named 'func_input' was not specified." in str(
e)
else:
assert False, 'Did not see expected exception!'
try:
# Bad gp name
f = simplestub.generate(target,
b_in,
float_arg=3.5,
offset=k,
func_input=f_in,
nonexistent_generator_param="wat")
except RuntimeError as e:
assert "Generator simplestub has no GeneratorParam named: nonexistent_generator_param" in str(
e)
else:
assert False, 'Did not see expected exception!'
def test_simplestub():
x, y = hl.Var(), hl.Var()
target = hl.get_jit_target_from_environment()
b_in = hl.Buffer(hl.UInt(8), [2, 2])
b_in.fill(123)
f_in = hl.Func("f")
f_in[x, y] = x + y
# ----------- Inputs by-position
f = simplestub.generate(target, b_in, f_in, 3.5)
_realize_and_check(f)
# ----------- Inputs by-name
f = simplestub.generate(target, buffer_input=b_in, func_input=f_in, float_arg=3.5)
_realize_and_check(f)
# ----------- Inputs w/ mixed by-position and by-name
f = simplestub.generate(target, b_in, f_in, float_arg=3.5)
_realize_and_check(f)
f = simplestub.generate(target, b_in, float_arg=3.5, func_input=f_in)
_realize_and_check(f)
# ----------- Above set again, w/ GeneratorParam mixed in
k = 42
f = simplestub.generate(target, b_in, f_in, 3.5, offset=k)
_realize_and_check(f, k)
f = simplestub.generate(target, offset=k, buffer_input=b_in, func_input=f_in, float_arg=3.5)
_realize_and_check(f, k)
f = simplestub.generate(target, b_in, f_in, offset=k, float_arg=3.5)
_realize_and_check(f, k)
f = simplestub.generate(target, b_in, float_arg=3.5, offset=k, func_input=f_in)
_realize_and_check(f, k)
# ----------- Test various failure modes
try:
# too many positional args
f = simplestub.generate(target, b_in, f_in, 3.5, 4)
except RuntimeError as e:
assert 'Expected at most 3 positional args, but saw 4.' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Inputs that can't be converted to what the receiver needs (positional)
f = simplestub.generate(target, 3.141592, "happy")
except RuntimeError as e:
assert 'Unable to cast Python instance' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Inputs that can't be converted to what the receiver needs (named)
f = simplestub.generate(target, b_in, f_in, float_arg="bogus")
except RuntimeError as e:
assert 'Unable to cast Python instance' in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Missing required inputs
f = simplestub.generate(target, b_in, f_in)
except RuntimeError as e:
assert "Generator Input named 'float_arg' was not specified." in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Input specified by both pos and kwarg
f = simplestub.generate(target, b_in, f_in, 3.5, float_arg=4.5)
except RuntimeError as e:
assert "Generator Input named 'float_arg' was specified by both position and keyword." in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Bad input name
f = simplestub.generate(target, b_in, float_arg=3.5, offset=k, funk_input=f_in)
except RuntimeError as e:
assert "Generator Input named 'func_input' was not specified." in str(e)
else:
assert False, 'Did not see expected exception!'
try:
# Bad gp name
f = simplestub.generate(target, b_in, float_arg=3.5, offset=k, func_input=f_in, nonexistent_generator_param="wat")
except RuntimeError as e:
assert "Generator has no GeneratorParam named: nonexistent_generator_param" in str(e)
else:
assert False, 'Did not see expected exception!'
