driver_api.h File Reference

Compiler driver APIs to drive the compilation. More...

#include <tvm/ir/global_var_supply.h>
#include <tvm/ir/module.h>
#include <tvm/ir/transform.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/support/with.h>
#include <tvm/target/target.h>
#include <tvm/te/schedule_pass.h>
#include <tvm/tir/function.h>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

Include dependency graph for driver_api.h:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.

Functions
transform::Sequential	tvm::MixedModulePassManager (IRModule mixed_mod, Target target)
	Configures and returns the composite Pass for the fused module (pre split) that contains device and host code. More...
transform::Sequential	tvm::DeviceModulePassManager (IRModule mixed_mod, Target target)
	Configures and returns the composite Pass for the device Target after device/host from mixed module. More...
transform::Sequential	tvm::HostModulePassManager (IRModule mixed_mod, Target target_host)
	Configures and returns the composite Pass for the host Target after device/host from mixed module. More...
IRModule	tvm::LowerModule (IRModule mod, bool simple_mode=false)
	Lower an IRModule (optimize with it with the pass list defined in CreatePassList) More...
IRModule	tvm::LowerPrimFunc (tvm::tir::PrimFunc func, const std::string &name, bool simple_mode=false)
	Lower a primfunc and name (convert to IRModule, and optimize it with the pass list defined in CreatePassList) More...
IRModule	tvm::LowerSchedule (te::Schedule sch, const Array< te::Tensor > &args, const std::string &name, const std::unordered_map< te::Tensor, tir::Buffer > &binds, GlobalVarSupply global_var_supply, bool simple_mode=false)
	Build an IRModule given a TE schedule, args and binds. This function also applies the lowering passes defined in CreatePassList. More...
IRModule	tvm::LowerSchedule (te::Schedule sch, const Array< ObjectRef > &args, const std::string &name, const std::unordered_map< te::Tensor, tir::Buffer > &binds, GlobalVarSupply global_var_supply, bool simple_mode=false)
	Build an IRModule given a TE schedule, args and binds. This function also applies the lowering passes defined in CreatePassList. More...
IRModule	tvm::ScheduleToModule (te::Schedule sch, const Array< ObjectRef > &args, const std::string &name, const std::unordered_map< te::Tensor, tir::Buffer > &binds, GlobalVarSupply global_var_supply)
	Create an IRModule out of a TE Schedule. It does not apply lowering passes. If you want to apply lowering passes as well, use LowerSchedule. More...
runtime::Module	tvm::build (const IRModule &funcs, const Target &target, const Target &target_host)
	Build a device and host module for a specific target from an IRModule. More...
runtime::Module	tvm::build (const Map< Target, IRModule > &input, const Target &target_host)
	Build a device and host module for a specific target from a map contains target to IRModule. This function is used for heterogeneous build. More...
runtime::Module	tvm::build (const Map< String, IRModule > &input, const Target &target_host)
	Build a device and host module for a specific target from a map contains target to IRModule. This function is used for heterogeneous build. More...

Detailed Description

Compiler driver APIs to drive the compilation.

This module provides end-to-end utils to drive the compilation process. We adopt the term "compiler driver" in common compiler infrastructures. Note that a compiler driver is different from "runtime drivers". Most of runtime related code are defined in the runtime folder instead.