python/unittest/test_arith_const_int_bound.py

25     bd = analyzer.const_int_bound(x)
26     assert bd.min_value == bd.NEG_INF
27     assert bd.max_value == bd.POS_INF
30     bd = analyzer.const_int_bound(x)
31     assert bd.min_value == -128
32     assert bd.max_value == 127
35     bd = analyzer.const_int_bound(x)
36     assert bd.min_value == 0
37     assert bd.max_value == 255
44     bd = analyzer.const_int_bound(tmod(x, 3).astype("uint32"))
45     assert bd.min_value == 0
46     assert bd.max_value == 2
48     bd = analyzer.const_int_bound(tmod(x, 3).astype("float32").astype("int32"))
49     assert bd.min_value == -2
50     assert bd.max_value == 2
56     bd = analyzer.const_int_bound(x + y)
57     assert bd.min_value == bd.NEG_INF
58     assert bd.max_value == bd.POS_INF
62     bd = analyzer.const_int_bound(x + y)
63     assert bd.min_value == 1
64     assert bd.max_value == 14
66     bd = analyzer.const_int_bound(x - y)
67     assert bd.min_value == -10
68     assert bd.max_value == 3
70     analyzer.update(x, tvm.arith.ConstIntBound(0, bd.POS_INF), override=True)
71     bd = analyzer.const_int_bound(x - y)
72     assert bd.min_value == -10
73     assert bd.max_value == bd.POS_INF
75     bd = analyzer.const_int_bound(1 - x)
76     assert bd.min_value == bd.NEG_INF
77     assert bd.max_value == 1
86     bd = analyzer.const_int_bound(x * y + 20)
87     assert bd.min_value == 0
88     assert bd.max_value == 60
92     bd = analyzer.const_int_bound(x * y)
93     assert bd.min_value == -32
94     assert bd.max_value == 24
96     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, 4), override=True)
98     bd = analyzer.const_int_bound(x * y)
99     assert bd.min_value == bd.NEG_INF
100     assert bd.max_value == bd.POS_INF
110     bd = analyzer.const_int_bound(tdiv(x, y))
111     assert bd.min_value == -2
115     bd = analyzer.const_int_bound(tdiv(x, y))
116     assert bd.min_value == -4
117     assert bd.max_value == 9
119     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, 4), override=True)
121     bd = analyzer.const_int_bound(tdiv(x, y))
122     assert bd.min_value == bd.NEG_INF
123     assert bd.max_value == bd.POS_INF
127     bd = analyzer.const_int_bound(tdiv(x, y))
128     assert bd.min_value == -9
129     assert bd.max_value == 9
140     bd = analyzer.const_int_bound(tmod(x, y))
141     assert bd.min_value == -9
142     assert bd.max_value == 4
144     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, bd.POS_INF), override=True)
146     bd = analyzer.const_int_bound(tmod(x, y))
147     assert bd.min_value == -9
148     assert bd.max_value == 9
150     analyzer.update(x, tvm.arith.ConstIntBound(1, bd.POS_INF), override=True)
152     bd = analyzer.const_int_bound(tmod(x, y))
153     assert bd.min_value == 0
154     assert bd.max_value == 9
163     bd = analyzer.const_int_bound(fld(x, y))
164     assert bd.min_value == -9 // 4
168     bd = analyzer.const_int_bound(fld(x, y))
169     assert bd.min_value == -4
170     assert bd.max_value == 9
172     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, 4), override=True)
174     bd = analyzer.const_int_bound(fld(x, y))
175     assert bd.min_value == bd.NEG_INF
176     assert bd.max_value == bd.POS_INF
180     bd = analyzer.const_int_bound(fld(x, y))
181     assert bd.min_value == -9
182     assert bd.max_value == 9
192     bd = analyzer.const_int_bound(flm(x, y))
193     assert bd.min_value == 0
194     assert bd.max_value == 9
196     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, bd.POS_INF), override=True)
198     bd = analyzer.const_int_bound(flm(x, y))
199     assert bd.min_value == 0
200     assert bd.max_value == 9
202     analyzer.update(x, tvm.arith.ConstIntBound(1, bd.POS_INF), override=True)
204     bd = analyzer.const_int_bound(flm(x, y))
205     assert bd.min_value == 0
206     assert bd.max_value == 9
215     bd = analyzer.const_int_bound(tvm.te.min(x, y))
216     assert bd.min_value == -9
217     assert bd.max_value == 10
219     analyzer.update(x, tvm.arith.ConstIntBound(bd.NEG_INF, bd.POS_INF), override=True)
221     bd = analyzer.const_int_bound(tvm.te.min(x, y))
222     assert bd.min_value == bd.NEG_INF
223     assert bd.max_value == 10
225     bd = analyzer.const_int_bound(tvm.te.max(x, y))
226     assert bd.min_value == 4
227     assert bd.max_value == bd.POS_INF
229     analyzer.update(x, tvm.arith.ConstIntBound(1, bd.POS_INF), override=True)
231     bd = analyzer.const_int_bound(tvm.te.max(x, y))
232     assert bd.min_value == 4
233     assert bd.max_value == bd.POS_INF
243     bd = analyzer.const_int_bound(tvm.tir.Select(x > 1, (y < 0).astype("int32"), y + 1))
244     assert bd.min_value == 0
245     assert bd.max_value == 11
255     bd = analyzer.const_int_bound(x >> y)
256     assert bd.min_value == -3
257     assert bd.max_value == 2
259     bd = analyzer.const_int_bound(x & y)
260     assert bd.min_value == 0
261     assert bd.max_value == 10
264     bd = analyzer.const_int_bound(x & y)
265     assert bd.min_value == 0
266     assert bd.max_value == 10
277     bd = analyzer.const_int_bound(tmod(x, 8) + tdiv(x, 8) * 8)
278     assert bd.min_value == 0
279     assert bd.max_value == 24 - 1
281     bd = analyzer.const_int_bound(y + x * 3)
282     assert bd.min_value == 0
283     assert bd.max_value == 24 * 3 - 1
285     bd = analyzer.const_int_bound(tmod(x, 7) + tdiv(x, 7) * 7)
286     assert bd.min_value == 0
287     assert bd.max_value == (23 // 7) * 7 + 6
293     bd = analyzer.const_int_bound(x)
294     assert bd.min_value == 0
295     assert bd.max_value == bd.POS_INF
301     bd = analyzer.const_int_bound(tvm.tir.Let(x, 1, x + 1))
302     assert bd.min_value == 2
303     assert bd.max_value == 2