[RFC] Meta Schedule (AutoTensorIR)

Development pre-RFC
#21

Hi there, how to apply the best schedule that meta schedule searched?

I’m now trying to apply like below:

    database = ms.database.JSONDatabase(f"{workdir}/database_workload.json",f"{workdir}/database_tuning_record.json")
    with ms.ApplyHistoryBest(database):
        with tvm.transform.PassContext(
            opt_level=3,
            config={"relay.backend.use_meta_schedule": True, "tir.predicate_opt": True},
        ):
            mod = tvm.build(mod, target="cuda")
            print(mod.imported_modules[0].get_source())

but it seems no schedule was applied to the irmodule.

#22

Please see compiler_tir if you’d like to build a single op

1 Like