Function save_object

tests/python/runtime/test_runtime_module_load.py:53–70

(names)

    temp = utils.tempdir()

    def save_object(names):
        n = te.size_var("n")
        Ab = tvm.tirx.decl_buffer((n,), dtype)
        i = te.var("i")
        # for i in 0 to n-1:
        stmt = tvm.tirx.For(
            i,
            0,
            n - 1,
            tvm.tirx.ForKind.SERIAL,
            tvm.tirx.BufferStore(Ab, tvm.tirx.BufferLoad(Ab, [i]) + 1, [i + 1]),
        )
        mod = tvm.IRModule.from_expr(
            tvm.tirx.PrimFunc([Ab], stmt).with_attr("global_symbol", "main")
        )
        m = tvm.tirx.build(mod, target=target)
        for name in names:
            m.write_to_file(name)

    path_obj = temp.relpath("test.o")
    path_ll = temp.relpath("test.ll")