tilelang.backend.target

Attributes

TargetConfig
TargetInput
TargetLike
TargetDetector
TargetNormalizer

Classes

TargetDetectorSpec
TargetNormalizerSpec

Functions

register_target_detector(name, detect, *[, override])
register_target_normalizer(name, normalize, *[, override])
auto_detect_target()
list_target_detectors()
determine_target([target, return_object])	Determine and validate the target for compilation.

Module Contents

tilelang.backend.target.TargetConfig

tilelang.backend.target.TargetInput

tilelang.backend.target.TargetLike

tilelang.backend.target.TargetDetector

tilelang.backend.target.TargetNormalizer

class tilelang.backend.target.TargetDetectorSpec

name: str

detect: TargetDetector

class tilelang.backend.target.TargetNormalizerSpec

name: str

normalize: TargetNormalizer

tilelang.backend.target.register_target_detector(name, detect, *, override=False)

Parameters:

• name (str)

• detect (TargetDetector)

• override (bool)

Return type:

TargetDetectorSpec

tilelang.backend.target.register_target_normalizer(name, normalize, *, override=False)

Parameters:

• name (str)

• normalize (TargetNormalizer)

• override (bool)

Return type:

TargetNormalizerSpec

tilelang.backend.target.auto_detect_target()

Return type:

TargetInput

tilelang.backend.target.list_target_detectors()

Return type:

tuple[str, Ellipsis]

tilelang.backend.target.determine_target(target='auto', return_object=False)

Determine and validate the target for compilation.

Target detection and normalization is provided by registered target modules.

Parameters:

• target (TargetLike | Literal['auto'])

• return_object (bool)

Return type:

str | collections.abc.Mapping[str, object] | tvm.target.Target