Function test_divide_by_zero

tests/python/tirx-base/test_tir_nodes.py:256–268

()

def test_divide_by_zero():
    for test in [
        lambda lhs, rhs: tvm.tirx.floormod(lhs, rhs),
        lambda lhs, rhs: tvm.tirx.floordiv(lhs, rhs),
        lambda lhs, rhs: tvm.tirx.truncmod(lhs, rhs),
        lambda lhs, rhs: tvm.tirx.truncdiv(lhs, rhs),
        lambda lhs, rhs: tvm.tirx.div(lhs, rhs),
    ]:
        try:
            test(tvm.tirx.const(5, "int32"), tvm.tirx.const(0, "int32"))
            assert False
        except RuntimeError:
            pass


def test_infinity():