Function build_tir_func

tests/python/tirx-base/test_tir_base.py:29–36

(func)

def build_tir_func(func):
    func = func.with_attr("global_symbol", "main")
    pass_ctx = PassContext.current()
    if pass_ctx.config.get("tirx.noalias", True):
        func = func.with_attr("tirx.noalias", True)
    mod = tvm.IRModule({"main": func})
    func = tvm.compile(mod)
    return func


def test_scalar_add():