Method main

tests/python/relax/test_transform_meta_schedule_apply_database.py:53–64

(A: T.Buffer((2,), "float32"), B: T.Buffer((2,), "float32"))

    class Expected:
        @T.prim_func(s_tir=True)
        def main(A: T.Buffer((2,), "float32"), B: T.Buffer((2,), "float32")):
            T.func_attr(
                {
                    "tirx.is_scheduled": True,
                    "global_symbol": "main",
                    "tirx.noalias": True,
                }
            )
            for i in T.serial(2):
                with T.sblock("renamed_block"):
                    vi = T.axis.spatial(2, i)
                    B[vi] = A[vi]

    def create_trace(mod: tvm.IRModule):
        sch = tvm.s_tir.Schedule(mod)