tvm_tirx.backend.cuda.BackendConfig

class tvm_tirx.backend.cuda.BackendConfig

Bases: TypedDict

Optional CUDA overrides. Toolchain arguments are passed as individual argv items.

arch: str

compiler: Literal['nvcc', 'nvrtc']

target_format: Literal['ptx', 'cubin', 'fatbin']

nvcc: list[str]

nvrtc: list[str]

ptxas: list[str]