| 22 | |
| 23 | |
| 24 | class IntSetChecker: |
| 25 | def __init__(self): |
| 26 | self.analyzer = tvm.arith.Analyzer() |
| 27 | |
| 28 | def verify(self, data, dmap, expected): |
| 29 | res = self.analyzer.int_set(data, dmap) |
| 30 | |
| 31 | def err_msg(): |
| 32 | return f"\ndata={data}\ndmap={dmap}\nres={res}\nexpected={expected}" |
| 33 | |
| 34 | assert self.analyzer.can_prove_equal(res.min_value, expected[0]), err_msg() |
| 35 | assert self.analyzer.can_prove_equal(res.max_value, expected[1]), err_msg() |
| 36 | |
| 37 | |
| 38 | def test_basic(): |
no outgoing calls
searching dependent graphs…