MCPcopy Create free account
hub / github.com/apache/tvm / codegen_build

Function codegen_build

python/tvm/tirx/build.py:122–130  ·  view source on GitHub ↗

Build a runtime module from an IRModule and a Target.

(mod: IRModule, target: Target)

Source from the content-addressed store, hash-verified

120
121
122def codegen_build(mod: IRModule, target: Target) -> tvm.runtime.Module:
123 """Build a runtime module from an IRModule and a Target."""
124 if tvm.ir.transform.PassContext.current().config.get("tirx.disable_assert", False):
125 mod = tvm.tirx.transform.SkipAssert()(mod)
126 build_f_name = "target.build." + target.kind.name
127 bf = tvm.get_global_func(build_f_name)
128 if bf is None:
129 raise ValueError(f"{build_f_name} is not enabled")
130 return bf(mod, target)
131
132
133def tir_to_runtime(

Callers 1

tir_to_runtimeFunction · 0.85

Calls 3

get_global_funcMethod · 0.80
getMethod · 0.45
currentMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…