tvm.driver

驱动程序API的命名空间

tvm.compile(mod: PrimFunc | IRModule, target: 目标 | None = None, *, relax_pipeline: 通过 | 可调用 | str | None = 'default', tir_pipeline: 通过 | 可调用 | str | None = 'default') → Executable

将IRModule编译为可运行时执行的模块。

该函数作为编译TIR和Relax模块的统一入口点。 它能自动检测模块类型并路由到相应的构建函数。

Parameters:

• mod (Union[PrimFunc, IRModule]) – 待编译的输入模块。可以是PrimFunc或包含TIR或Relax函数的IRModule。

• target (可选[Target]) – 要编译的目标平台。

• relax_pipeline (可选[联合类型[tvm.transform.Pass, 可调用对象, 字符串]]) - 用于Relax函数的编译流水线。 仅当模块包含Relax函数时使用。

• tir_pipeline (Optional[Union[tvm.transform.Pass, Callable, str]]) – 用于TIR函数的编译流水线。

Returns:

一个可加载和执行的运行时可执行文件。

Return type:

可执行