-
Notifications
You must be signed in to change notification settings - Fork 57
/
test_frontend_op_resolver.py
79 lines (62 loc) · 2.5 KB
/
test_frontend_op_resolver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import pytest
import xdsl.frontend.dialects.builtin as builtin
from xdsl.dialects.arith import Addi, Constant, Mulf
from xdsl.dialects.builtin import (
Float32Type,
Float64Type,
FloatAttr,
IntegerType,
f32,
f64,
i32,
i64,
)
from xdsl.frontend.exception import FrontendProgramException
from xdsl.frontend.op_resolver import OpResolver
def test_raises_exception_on_unknown_op():
with pytest.raises(FrontendProgramException) as err:
_ = OpResolver.resolve_op("xdsl.frontend.dialects.arith", "unknown")
assert (
err.value.msg
== "Internal failure: operation 'unknown' does not exist in module 'xdsl.frontend.dialects.arith'."
)
def test_raises_exception_on_unknown_overload():
with pytest.raises(FrontendProgramException) as err:
_ = OpResolver.resolve_op_overload("__unknown__", builtin._Integer)
assert (
err.value.msg == "Internal failure: '_Integer' does not overload '__unknown__'."
)
def test_resolves_ops():
a = Constant.from_int_and_width(1, i32)
b = Constant.from_int_and_width(2, i32)
addi = OpResolver.resolve_op("xdsl.frontend.dialects.arith", "addi")
addi_op = addi(a, b)
assert isinstance(addi_op, Addi)
assert addi_op.operands[0] == a.results[0]
assert addi_op.operands[1] == b.results[0]
assert isinstance(addi_op.results[0].type, IntegerType)
assert addi_op.results[0].type.width.data == 32
c = Constant(FloatAttr(5.0, f32))
mulf = OpResolver.resolve_op("xdsl.frontend.dialects.arith", "mulf")
mulf_op = mulf(c, c)
assert isinstance(mulf_op, Mulf)
assert mulf_op.operands[0] == c.results[0]
assert mulf_op.operands[1] == c.results[0]
assert isinstance(mulf_op.results[0].type, Float32Type)
def test_resolves_overloads():
a = Constant.from_int_and_width(1, i64)
b = Constant.from_int_and_width(2, i64)
addi = OpResolver.resolve_op_overload("__add__", builtin._Integer)
addi_op = addi(a, b)
assert isinstance(addi_op, Addi)
assert addi_op.operands[0] == a.results[0]
assert addi_op.operands[1] == b.results[0]
assert isinstance(addi_op.results[0].type, IntegerType)
assert addi_op.results[0].type.width.data == 64
c = Constant(FloatAttr(5.0, f64))
mulf = OpResolver.resolve_op_overload("__mul__", builtin._Float64)
mulf_op = mulf(c, c)
assert isinstance(mulf_op, Mulf)
assert mulf_op.operands[0] == c.results[0]
assert mulf_op.operands[1] == c.results[0]
assert isinstance(mulf_op.results[0].type, Float64Type)