Tianqi Chen
706f9b6f7e
* [CODEGEN/RUNTIME] Metal support, runtime improvement. * Fix case when no device is available
23 lines
453 B
Python
23 lines
453 B
Python
"""Code generation related functions."""
|
|
from ._ffi.function import _init_api
|
|
|
|
def build_module(lowered_func, target):
|
|
"""Build lowered_func into Module.
|
|
|
|
Parameters
|
|
----------
|
|
lowered_func : LoweredFunc
|
|
The lowered function
|
|
|
|
target : str
|
|
The target module type.
|
|
|
|
Returns
|
|
-------
|
|
module : Module
|
|
The corressponding module.
|
|
"""
|
|
return _Build(lowered_func, target)
|
|
|
|
_init_api("tvm.codegen")
|