tilelang.jit.adapter.nvrtc

NVRTC Backend for TileLang.

This module provides runtime compilation support using NVIDIA's NVRTC API.

Submodules

• tilelang.jit.adapter.nvrtc.adapter
• tilelang.jit.adapter.nvrtc.kernel_cache
• tilelang.jit.adapter.nvrtc.libgen
• tilelang.jit.adapter.nvrtc.wrapper

Attributes

is_nvrtc_available
is_nvrtc_available

Classes

NVRTCKernelAdapter

Dummy NVRTCKernelAdapter that raises ImportError on instantiation.

Functions

check_nvrtc_available()

Check if NVRTC backend is available.

Package Contents

tilelang.jit.adapter.nvrtc.is_nvrtc_available = False

tilelang.jit.adapter.nvrtc.is_nvrtc_available = True

tilelang.jit.adapter.nvrtc.check_nvrtc_available()

Check if NVRTC backend is available.

抛出:

ImportError -- If cuda-python is not installed or cannot be imported

class tilelang.jit.adapter.nvrtc.NVRTCKernelAdapter(*args, **kwargs)

Dummy NVRTCKernelAdapter that raises ImportError on instantiation.

classmethod from_database(*args, **kwargs)