Python extension module compiler

class returnn.util.py_ext_mod_compiler.PyExtModCompiler(include_paths=(), **kwargs)[source]

Python extension module compiler

  • base_name (str) – base name for the module, e.g. “zero_out”

  • code_version (int|tuple[int]) – check for the cache whether to reuse

  • code (str) – the source code itself

  • is_cpp (bool) – if False, C is assumed

  • c_macro_defines (dict[str,str|int|None]|None) – e.g. {“TENSORFLOW”: 1}

  • ld_flags (list[str]|None) – e.g. [“-lblas”]

  • include_paths (list[str]|tuple[str])

  • include_deps (list[str]|None) – if provided and an existing lib file, we will check if any dependency is newer and we need to recompile. we could also do it automatically via -MD but that seems overkill and too slow.

  • static_version_name (str|None) – normally, we use …/base_name/hash as the dir but this would use …/base_name/static_version_name.

  • should_cleanup_old_all (bool) – whether we should look in the cache dir and check all ops if we can delete some old ones which are older than some limit (self._cleanup_time_limit_days)

  • should_cleanup_old_mydir (bool) – whether we should delete our op dir before we compile there.

  • log_stream (TextIO|None) – file stream for print statements

  • verbose (bool) – be slightly more verbose

CacheDirName = 'returnn_py_ext_mod_cache'[source]

Python extension module