Function test_buffer_replacer_no_shared_default

tests/python/tirx/test_op.py:47–57

Regression test for F4: BufferReplacer default dicts must not be shared.

()

def test_buffer_replacer_no_shared_default():
    """Regression test for F4: BufferReplacer default dicts must not be shared."""
    from tvm.tirx.transform.common import BufferReplacer

    r1 = BufferReplacer()
    r2 = BufferReplacer()
    A = decl_buffer((64,), "float32")
    B = decl_buffer((64,), "float32")
    r1.buffer_map[A] = B
    # r2 must not see r1's mutation
    assert len(r2.buffer_map) == 0


def test_gemm_async_partial_scale_factor():