tilelang.contrib.cc¶

Util to invoke C/C++ compilers in the system.

Functions¶

`get_cc`()	Return the path to the default C/C++ compiler.
`get_cplus_compiler`()	Return the path to the default C/C++ compiler.
`create_shared`(output, objects[, options, cc, cwd, ...])	Create shared library.
`create_staticlib`(output, inputs[, ar])	Create static library.
`create_executable`(output, objects[, options, cc, cwd, ...])	Create executable binary.
`get_global_symbol_section_map`(path, *[, nm])	Get global symbols from a library via nm -g
`get_target_by_dump_machine`(compiler)	Functor of get_target_triple that can get the target triple using compiler.
`cross_compiler`(compile_func[, options, output_format, ...])	Create a cross compiler function by specializing compile_func with options.

Module Contents¶

tilelang.contrib.cc.get_cc()¶

Return the path to the default C/C++ compiler.

Returns:: out – The path to the default C/C++ compiler, or None if none was found.
Return type:: Optional[str]

tilelang.contrib.cc.get_cplus_compiler()¶

Return the path to the default C/C++ compiler.

Returns:: out – The path to the default C/C++ compiler, or None if none was found.
Return type:: Optional[str]

tilelang.contrib.cc.create_shared(output, objects, options=None, cc=None, cwd=None, ccache_env=None)¶

Create shared library.

Parameters:

output (str) – The target shared library.
objects (List[str]) – List of object files.
options (List[str]) – The list of additional options string.
cc (Optional[str]) – The compiler command.
cwd (Optional[str]) – The current working directory.
ccache_env (Optional[Dict[str, str]]) – The environment variable for ccache. Set None to disable ccache by default.

tilelang.contrib.cc.create_staticlib(output, inputs, ar=None)¶

Create static library.

Parameters:

output (str) – The target shared library.
inputs (List[str]) – List of inputs files. Each input file can be a tarball of objects or an object file.
ar (Optional[str]) – Path to the ar command to be used

tilelang.contrib.cc.create_executable(output, objects, options=None, cc=None, cwd=None, ccache_env=None)¶

Create executable binary.

Parameters:

output (str) – The target executable.
objects (List[str]) – List of object files.
options (List[str]) – The list of additional options string.
cc (Optional[str]) – The compiler command.
cwd (Optional[str]) – The urrent working directory.
ccache_env (Optional[Dict[str, str]]) – The environment variable for ccache. Set None to disable ccache by default.

tilelang.contrib.cc.get_global_symbol_section_map(path, *, nm=None)¶

Get global symbols from a library via nm -g

Parameters:

path (str) – The library path
nm (str) – The path to nm command

Returns:

symbol_section_map – A map from defined global symbol to their sections

Return type:

Dict[str, str]

tilelang.contrib.cc.get_target_by_dump_machine(compiler)¶

Functor of get_target_triple that can get the target triple using compiler.

Parameters:: compiler (Optional[str]) – The compiler.
Returns:: out – A function that can get target triple according to dumpmachine option of compiler.
Return type:: Callable

tilelang.contrib.cc.cross_compiler(compile_func, options=None, output_format=None, get_target_triple=None, add_files=None)¶

Create a cross compiler function by specializing compile_func with options.

This function can be used to construct compile functions that can be passed to AutoTVM measure or export_library.

Parameters:

compile_func (Union[str, Callable[[str, str, Optional[str]], None]]) – Function that performs the actual compilation
options (Optional[List[str]]) – List of additional optional string.
output_format (Optional[str]) – Library output format.
get_target_triple (Optional[Callable]) – Function that can target triple according to dumpmachine option of compiler.
add_files (Optional[List[str]]) – List of paths to additional object, source, library files to pass as part of the compilation.

Returns:

fcompile – A compilation function that can be passed to export_library.

Return type:

Callable[[str, str, Optional[str]], None]

Examples

from tvm.contrib import cc, ndk
# export using arm gcc
mod = build_runtime_module()
mod.export_library(path_dso,
                   fcompile=cc.cross_compiler("arm-linux-gnueabihf-gcc"))
# specialize ndk compilation options.
specialized_ndk = cc.cross_compiler(
    ndk.create_shared,
    ["--sysroot=/path/to/sysroot", "-shared", "-fPIC", "-lm"])
mod.export_library(path_dso, fcompile=specialized_ndk)