tilelang.backend.execution_backend¶

Attributes¶

TargetPredicate

AvailabilityCheck

Classes¶

ExecutionBackendSpec

Functions¶

canonicalize_execution_backend(name)

register_execution_backend(target_kind, spec, *[, ...])

register_lazy_execution_backends(target_kind, import_path)

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

resolve_execution_backend(requested, target)

resolve_execution_backend_spec(requested, target)

Module Contents¶

tilelang.backend.execution_backend.TargetPredicate¶
tilelang.backend.execution_backend.AvailabilityCheck¶
tilelang.backend.execution_backend.canonicalize_execution_backend(name)¶
Parameters:

name (str | None)

Return type:

str | None

class tilelang.backend.execution_backend.ExecutionBackendSpec¶
name: str¶
is_available: AvailabilityCheck¶
supports_target: TargetPredicate | None = None¶
enable_host_codegen: bool = False¶
enable_device_compile: bool = False¶
matches(target)¶
Parameters:

target (tvm.target.Target)

Return type:

bool

tilelang.backend.execution_backend.register_execution_backend(target_kind, spec, *, override=False)¶
Parameters:
Return type:

ExecutionBackendSpec

tilelang.backend.execution_backend.register_lazy_execution_backends(target_kind, import_path)¶
Parameters:

  • target_kind (str)

  • import_path (str)

Return type:

None

tilelang.backend.execution_backend.allowed_backends_for_target(target, *, include_unavailable=True)¶
Parameters:

  • target (tvm.target.Target)

  • include_unavailable (bool)

Return type:

list[str]

tilelang.backend.execution_backend.resolve_execution_backend(requested, target)¶
Parameters:

  • requested (str | None)

  • target (tvm.target.Target)

Return type:

str

tilelang.backend.execution_backend.resolve_execution_backend_spec(requested, target)¶
Parameters:

  • requested (str | None)

  • target (tvm.target.Target)

Return type:

ExecutionBackendSpec