Function test_op_attrs

tests/python/relax/test_ast_printer.py:419–434

()

def test_op_attrs():
    x = rx.Var("x", R.Tensor((10,), "float32"))
    # Manually create a Call with attributes to test printer support for Op attributes
    op = tvm.ir.Op.get("relax.add")
    attrs = tvm.ir.make_node("ir.DictAttrs", my_attr="my_value")
    call_node = rx.Call(op, [x, x], attrs=attrs)

    call_str = dump_ast(call_node, include_call_attrs=True)
    assert_fields(
        "Call",
        {
            "op": 'Op(name="relax.add")',
            "attrs": '{"my_attr": "my_value"}',
        },
        call_str,
    )


def test_call_tir():