tilelang.contrib.nvrtc ====================== .. py:module:: tilelang.contrib.nvrtc Functions --------- .. autoapisummary:: tilelang.contrib.nvrtc.get_nvrtc_version tilelang.contrib.nvrtc.compile_cuda Module Contents --------------- .. py:function:: get_nvrtc_version() .. py:function:: compile_cuda(code, target_format = 'ptx', arch = None, options = None, verbose = False) Compile cuda code with NVRTC. :param code: The cuda code. :type code: str :param target_format: The target format of nvrtc compiler. :type target_format: Literal["ptx", "cubin"] :param arch: The cuda architecture code. :type arch: Optional[int] :param options: The additional options. :type options: Optional[Union[str, List[str]]] :param verbose: Whether to print the verbose output. :type verbose: bool :returns: **result_bytes** -- The bytearray of the cubin or ptx code. :rtype: bytearray