[VTA] Where can I find codegen part?

#1

Hello, everyone.

From deploy_vision_model.py,

with relay.build_config(opt_level=3, disabled_pass={"AlterOpLayout"}):
    if target.device_name != "vta":
        graph, lib, params = relay.build(
            relay_prog, target=target,
            params=params, target_host=env.target_host)
    else:
        with vta.build_config():
            graph, lib, params = relay.build(
                relay_prog, target=target,
                params=params, target_host=env.target_host)

out_file = open("inf.ll", "w")
out_file.write(lib.get_source())
out_file.close()

Since out_file shows llvm .ll file, I’m sure that relay.build has a part that is related to LLVM code generation.

I found the lowering part but can’t find the part that leads to codegen_llvm or those related.

Can anyone tell me where to look up on?

Thanks,

Jake