def test_mismatch_type_mod(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([pt.Return(pt.Mod(pt.Int8(2), pt.Int32(2)))]),
)
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([pt.Return(pt.Mod(pt.Int32(2), pt.Int8(2)))]),
)
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int64_t,
pt.Block([pt.Return(pt.Mod(pt.Int16(2), pt.Int64(2)))]),
)
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int64_t,
pt.Block([pt.Return(pt.Mod(pt.Int64(2), pt.Int16(2)))]),
)
def test_return_call_nothing_for_int32_t(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"nothing", (), (), pt.Block([pt.Return(())])).add_func(
"call_nothing",
(),
pt.Int32_t,
pt.Block([pt.Return(pt.Call("nothing", []))]),
)
def test_mismatch_type_deref(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([pt.Return(pt.Deref(pt.Int64(123)))]),
)
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo", (ptr, ), (),
pt.Block([
pt.DefineVar(v32, pt.Deref(pt.Var(ptr))),
]))
def test_assign_undeclared_variable(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([
pt.Assign(pt.Var(x), pt.Int32(500)),
pt.Return(pt.Int32(2)),
]),
)
def test_redeclared_variable(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([
pt.DefineVar(x, pt.Int32(2)),
pt.DefineVar(x, pt.Int32(3)),
pt.Return(pt.Int32(2)),
]),
)
def test_declare_wrong_type(self) -> None:
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([
pt.DefineVar(z, pt.Int32(2)),
pt.Return(pt.Int32(2)),
]),
)
with self.assertRaises(pt.TypeCheckError):
pt.Program("module").add_func(
"foo",
(),
pt.Int32_t,
pt.Block([
pt.DefineVar(x, pt.Int8(2)),
pt.Return(pt.Int32(2)),
]),
)
import petra as pt
import unittest
from ctypes import CFUNCTYPE, c_int32, c_bool
program = pt.Program("module")
program.add_func(
"if_else_true",
(),
pt.Int32_t,
pt.Block([
pt.If(
pt.Lt(pt.Int32(0), pt.Int32(1)),
pt.Block([
pt.Return(pt.Int32(1)),
]),
pt.Block([pt.Return(pt.Int32(2))]),
),
pt.Return(pt.Int32(0)),
]),
)
program.add_func(
"if_else_false",
(),
pt.Int32_t,
pt.Block([
pt.If(
pt.Lt(pt.Int32(1), pt.Int32(0)),
pt.Block([
pt.Return(pt.Int32(1)),
pt.Block([
pt.DefineVar(a, pt.Add(pt.Int8(-11), pt.Int8(-4))),
pt.DefineVar(b, pt.Add(pt.Int8(-11), pt.Int8(0))),
pt.DefineVar(c, pt.Add(pt.Int8(-11), pt.Int8(7))),
pt.DefineVar(d, pt.Add(pt.Int8(0), pt.Int8(-5))),
pt.DefineVar(e, pt.Add(pt.Int8(0), pt.Int8(0))),
pt.DefineVar(f, pt.Add(pt.Int8(0), pt.Int8(3))),
pt.DefineVar(g, pt.Add(pt.Int8(7), pt.Int8(-8))),
pt.DefineVar(h, pt.Add(pt.Int8(7), pt.Int8(0))),
pt.DefineVar(i, pt.Add(pt.Int8(7), pt.Int8(8))),
pt.Return(
pt.Add(
pt.Var(a),
pt.Add(
pt.Var(b),
pt.Add(
pt.Var(c),
pt.Add(
pt.Var(d),
pt.Add(
pt.Var(e),
pt.Add(
pt.Var(f),
pt.Add(
pt.Var(g),
pt.Add(pt.Var(h), pt.Var(i)),
),
),
),
),
),
),
)),
]),
program = pt.Program("module")
n = pt.Symbol(pt.Int32_t, "n")
program.add_func(
"collatz",
(n, ),
pt.Int32_t,
pt.Block([
pt.If(
pt.Eq(pt.Var(n), pt.Int32(1)),
pt.Block([pt.Return(pt.Int32(0))]),
pt.Block([]),
),
pt.If(
pt.Eq(pt.Mod(pt.Var(n), pt.Int32(2)), pt.Int32(0)),
pt.Block([pt.Assign(pt.Var(n), pt.Div(pt.Var(n), pt.Int32(2)))]),
pt.Block([
pt.Assign(
pt.Var(n),
pt.Add(pt.Mul(pt.Int32(3), pt.Var(n)), pt.Int32(1)),
),
]),
),
pt.Return(pt.Add(pt.Int32(1), pt.Call("collatz", [pt.Var(n)]))),
]),
)
program.save_object("collatz.py.o")
from typing import cast, Callable
import petra as pt
import unittest
from ctypes import CFUNCTYPE, c_int8, c_int16, c_int32, c_int64
program = pt.Program("module")
# Int8_t functions.
program.add_func("return_m2_i8", (), pt.Int8_t,
pt.Block([pt.Return(pt.Int8(-2))]))
program.add_func("return_0_i8", (), pt.Int8_t,
pt.Block([pt.Return(pt.Int8(0))]))
program.add_func("return_2_i8", (), pt.Int8_t,
pt.Block([pt.Return(pt.Int8(2))]))
# Int16_t functions.
program.add_func("return_m2_i16", (), pt.Int16_t,
pt.Block([pt.Return(pt.Int16(-2))]))
program.add_func("return_0_i16", (), pt.Int16_t,
pt.Block([pt.Return(pt.Int16(0))]))
program.add_func("return_2_i16", (), pt.Int16_t,
pt.Block([pt.Return(pt.Int16(2))]))
program.add_func(
"array_set_get_values",
(),
pt.Int32_t,
pt.Block(
[
pt.DefineVar(array_var),
pt.Assign(
pt.Var(array_var), pt.SetElement(pt.Var(array_var), pt.Int32(1), 0)
),
pt.Assign(
pt.Var(array_var), pt.SetElement(pt.Var(array_var), pt.Int32(2), 1)
),
pt.Assign(
pt.Var(array_var), pt.SetElement(pt.Var(array_var), pt.Int32(3), 2)
),
pt.Return(
pt.Add(
pt.GetElement(pt.Var(array_var), 0),
pt.Add(
pt.GetElement(pt.Var(array_var), 1),
pt.GetElement(pt.Var(array_var), 2),
),
)
),
]
),
)
class ArraysTestCase(unittest.TestCase):
import petra as pt
import unittest
from ctypes import CFUNCTYPE, c_bool, c_int8
program = pt.Program("module")
x, y = pt.Symbol(pt.Int8_t, "x"), pt.Symbol(pt.Int8_t, "y")
z, w = pt.Symbol(pt.Bool_t, "z"), pt.Symbol(pt.Bool_t, "w")
program.add_func(
"lt",
(x, y),
pt.Bool_t,
pt.Block([pt.Return(pt.Lt(pt.Var(x), pt.Var(y)))]),
)
program.add_func(
"lte",
(x, y),
pt.Bool_t,
pt.Block([pt.Return(pt.Lte(pt.Var(x), pt.Var(y)))]),
)
program.add_func(
"gt",
(x, y),
pt.Bool_t,
pt.Block([pt.Return(pt.Gt(pt.Var(x), pt.Var(y)))]),
)
import petra as pt
program = pt.Program("module")
program.add_func_decl("sqrtf", (pt.Float32_t, ), pt.Float32_t)
x = pt.Symbol(pt.Float32_t, "x")
program.add_func(
"call_sqrtf",
(x, ),
pt.Float32_t,
pt.Block([pt.Return(pt.Call("sqrtf", [pt.Var(x)]))]),
)
program.save_object("sqrt.py.o")
import petra as pt
import unittest
from ctypes import CFUNCTYPE, c_int32
program = pt.Program("module")
x = pt.Symbol(pt.Int32_t, "x")
y = pt.Symbol(pt.Int32_t, "y")
z = pt.Symbol(pt.Int8_t, "z")
program.add_func(
"return_temp",
(),
pt.Int32_t,
pt.Block([pt.DefineVar(x, pt.Int32(2)),
pt.Return(pt.Var(x))]),
)
program.add_func(
"return_shuffle_temp",
(),
pt.Int32_t,
pt.Block([
pt.DefineVar(x, pt.Int32(2)),
pt.DefineVar(y, pt.Var(x)),
pt.Return(pt.Var(y)),
]),
)
program.add_func(
"temp_unused",
pt.Block([
pt.DefineVar(
point1d,
pt.Call("legion_domain_point_get_point_1d",
[pt.Deref(pt.Var(point_ptr))])),
pt.DefineVar(
point1d_x,
pt.GetElement(pt.Var(point1d), name="x"),
),
pt.DefineVar(x, pt.GetElement(pt.Var(point1d_x), idx=0)),
pt.DefineVar(x_plus_1, pt.Add(pt.Var(x), pt.Int64(1))),
pt.DefineVar(point1d_x_plus_1_x),
pt.Assign(
pt.Var(point1d_x_plus_1_x),
pt.SetElement(pt.Var(point1d_x_plus_1_x), pt.Var(x_plus_1), 0),
),
pt.DefineVar(point1d_x_plus_1),
pt.Assign(
pt.Var(point1d_x_plus_1),
pt.SetElement(pt.Var(point1d_x_plus_1),
pt.Var(point1d_x_plus_1_x),
name="x"),
),
pt.DefineVar(
domain_point_x_plus_1,
pt.Call("legion_domain_point_from_point_1d",
[pt.Var(point1d_x_plus_1)]),
),
pt.DefineVar(
result,
pt.Call(
"legion_logical_partition_get_logical_subregion_by_color_domain_point",
[
pt.Var(runtime),
pt.Deref(pt.Var(parent_ptr)),
pt.Var(domain_point_x_plus_1)
],
),
),
pt.Return(pt.Var(result)),
]),
program.add_func_decl("free", (Pointer_Int64_t, ), ())
# void* memset ( void* ptr, int value, size_t num );
program.add_func_decl("memset", (Pointer_Int64_t, pt.Int32_t, pt.Int64_t),
Pointer_Int64_t)
ptr = pt.Symbol(Pointer_Int64_t, "ptr")
val = pt.Symbol(pt.Int64_t, "val")
v32 = pt.Symbol(pt.Int32_t, "val")
program.add_func(
"malloc_free",
(),
pt.Int32_t,
pt.Block([
pt.DefineVar(ptr, pt.Call("malloc", [pt.Int64(8)])),
pt.Call("free", [pt.Var(ptr)]),
pt.Return(pt.Int32(1)),
]),
)
program.add_func(
"malloc_memset",
(),
pt.Int64_t,
pt.Block([
pt.DefineVar(ptr, pt.Call("malloc", [pt.Int64(8)])),
pt.Assign(
pt.Var(ptr),
pt.Call("memset",
[pt.Var(ptr), pt.Int32(0x3A),
pt.Int64(8)]),
from typing import cast, Callable
import petra as pt
import unittest
from ctypes import CFUNCTYPE, c_int32
program = pt.Program("module")
x, y = pt.Symbol(pt.Int32_t, "x"), pt.Symbol(pt.Int32_t, "y")
program.add_func("return_2", (), pt.Int32_t,
pt.Block([pt.Return(pt.Int32(2))]))
program.add_func("call_return_2", (), pt.Int32_t,
pt.Block([pt.Return(pt.Call("return_2", []))]))
program.add_func(
"call_return_2_discard",
(),
pt.Int32_t,
pt.Block([pt.Call("return_2", []),
pt.Return(pt.Int32(3))]),
)
program.add_func(
"iden",
(x, ),
pt.Int32_t,
pt.Block([pt.Return(pt.Var(x))]),
)