tilelang.contrib.nvrtc

Functions

get_nvrtc_version()

compile_cuda(code[, target_format, arch, options, verbose])

Compile cuda code with NVRTC.

Module Contents

tilelang.contrib.nvrtc.get_nvrtc_version()
返回类型:

tuple[int, int]

tilelang.contrib.nvrtc.compile_cuda(code, target_format='ptx', arch=None, options=None, verbose=False)

Compile cuda code with NVRTC.

参数:

  • code (str) -- The cuda code.

  • target_format (Literal["ptx", "cubin"]) -- The target format of nvrtc compiler.

  • arch (Optional[int]) -- The cuda architecture code.

  • options (Optional[Union[str, List[str]]]) -- The additional options.

  • verbose (bool) -- Whether to print the verbose output.

返回:

result_bytes -- The bytearray of the cubin or ptx code.

返回类型:

bytearray