tilelang.contrib.cuda_include¶

CUDA include-path discovery shared by compiler integrations.

Functions¶

discover_cuda_include_paths(cuda_home[, machine, system])

Return NVRTC include paths supported by the CUDA layout at cuda_home.

Module Contents¶

tilelang.contrib.cuda_include.discover_cuda_include_paths(cuda_home, machine=None, system=None)¶

Return NVRTC include paths supported by the CUDA layout at cuda_home.

CUDA pip packages may install headers in a flat include tree, while system toolkits commonly use targets/<arch>-linux/include. Include both layouts when present so split or overlaid installations continue to work.

Parameters:

  • cuda_home (str)

  • machine (str | None)

  • system (str | None)

Return type:

list[str]