d9cecdf50141cc56eec9be928063a81baba31d92,tests/python/unittest/test_tir_buffer.py,,test_buffer_access_ptr_offset,#,46

Before Change

    assert tvm.ir.structural_equal(offset, 200 + v)
    assert aptr.args[4].value == Buffer.READ | Buffer.WRITE
    aptr = Ab.access_ptr("rw", offset=tvm.tir.call_extern("int32", "test_call", 100 + 100 + v))
    offset = tvm.tir.ir_pass.Simplify(aptr.args[2])
    assert tvm.ir.structural_equal(offset, tvm.tir.call_extern("int32", "test_call", 200 + v))
    assert aptr.args[4].value == Buffer.READ | Buffer.WRITE

After Change

    tvm.testing.assert_prim_expr_equal(aptr.args[2], 200 + v)
    assert aptr.args[4].value == Buffer.READ | Buffer.WRITE
    aptr = Ab.access_ptr("rw", offset=tvm.tir.call_extern("int32", "test_call", 100 + 100 + v))
    tvm.testing.assert_prim_expr_equal(aptr.args[2], tvm.tir.call_extern("int32", "test_call", 200 + v))
    assert aptr.args[4].value == Buffer.READ | Buffer.WRITE


def test_buffer_access_ptr_extent():

Instances