tilelang.backend.target¶

Attributes¶

`TargetConfig`
`TargetInput`
`TargetLike`
`TargetDetector`
`TargetNormalizer`
`TargetExecutionNormalizer`

Classes¶

`TargetDetectorSpec`
`TargetNormalizerSpec`
`TargetExecutionNormalizerSpec`

Functions¶

`register_target_detector`(name, detect, *[, override])
`register_target_normalizer`(name, normalize, *[, override])
`register_target_execution_normalizer`(name, normalize, *)	Register target specialization driven by an execution backend choice.
`normalize_target_for_execution`(target, execution_backend)	Apply the first registered execution-aware target specialization.
`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¶

tilelang.backend.target.TargetExecutionNormalizer¶

class tilelang.backend.target.TargetDetectorSpec¶

name: str¶

detect: TargetDetector¶

class tilelang.backend.target.TargetNormalizerSpec¶

name: str¶

normalize: TargetNormalizer¶

class tilelang.backend.target.TargetExecutionNormalizerSpec¶

name: str¶

normalize: TargetExecutionNormalizer¶

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.register_target_execution_normalizer(name, normalize, *, override=False)¶

Parameters:

name (str)
normalize (TargetExecutionNormalizer)
override (bool)

Return type:

TargetExecutionNormalizerSpec

tilelang.backend.target.normalize_target_for_execution(target, execution_backend)¶

Apply the first registered execution-aware target specialization.

Parameters:

target (tvm.target.Target)
execution_backend (str | None)

Return type:

tvm.target.Target

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