Build a runtime module from an IRModule and a Target.
(mod: IRModule, target: Target)
| 120 | |
| 121 | |
| 122 | def 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 | |
| 133 | def tir_to_runtime( |
no test coverage detected
searching dependent graphs…