tilelang.jit.execution_backend

Functions

allowed_backends_for_target(target, *[, ...])

Return allowed execution backends for a given TVM target kind.

resolve_execution_backend(requested, target)

Resolve an execution backend string to a concrete backend.

Module Contents

tilelang.jit.execution_backend.allowed_backends_for_target(target, *, include_unavailable=True)

Return allowed execution backends for a given TVM target kind.

include_unavailable: if False, this will filter out backends that are known to be unavailable at runtime (e.g., NVRTC without cuda-python installed).

Parameters:

  • target (tvm.target.Target)

  • include_unavailable (bool)

Return type:

list[str]

tilelang.jit.execution_backend.resolve_execution_backend(requested, target)

Resolve an execution backend string to a concrete backend.

  • Supports the alias “dlpack” -> “tvm_ffi”.

  • Supports the sentinel “auto” which selects a sensible default per target.

  • Validates the combination (target, backend) and raises with helpful alternatives when invalid.

Parameters:

  • requested (str | None)

  • target (tvm.target.Target)

Return type:

str