Function test_bind

tests/python/tvmscript/test_tvmscript_printer_tir.py:270–288

()

def test_bind():
    with IRBuilder() as ib:
        with T.prim_func(s_tir=True):
            v = T.bind(T.float32(10))
            ib.name("v", v)
            T.evaluate(1)
    obj = ib.get()
    _assert_print(
        obj,
        """
# from tvm.script import tirx as T
# from tvm.tirx.layout import Axis

@T.prim_func(private=True, s_tir=True)
def main():
    v: T.let[T.float32] = T.float32(10.0)
    T.evaluate(1)
""",
    )


def test_attr_stmt():