Function operation

tests/python/tvmscript/test_tvmscript_parser_tir.py:646–649

(A, idx)

def test_tir_macro_block_name_suffix():
    @T.inline
    def operation(A, idx):
        with T.sblock("op"):
            v = T.axis.remap("S", [idx])
            A[v] = A[v] * T.float32(2)

    @T.prim_func(private=True, s_tir=True)
    def func_w_macro(a: T.handle) -> None: