tvm.relax.backend#

Relax backends

class tvm.relax.backend.DispatchSampling(*args, **kwargs)#

Pass to dispatch scan and sort operators to platform dependent implementation.

class tvm.relax.backend.DispatchSortScan(*args, **kwargs)#

Pass to dispatch scan and sort operators to platform dependent implementation.

tvm.relax.backend.get_pattern(name: str) → FusionPattern | None#

Find the pattern with a particular name.

Parameters:

name (str) – The pattern name.

Returns:

pattern – The matched pattern. Returns None if such pattern is not found.

Return type:

Optional[FusionPattern]

tvm.relax.backend.get_patterns_with_prefix(prefix: str) → list[FusionPattern]#

Get a list of patterns whose names startwith prefix.

Parameters:

prefix (str) – The prefix of pattern name.

Returns:

patterns – Matched patterns, ordered by priority from high to low.

Return type:

FusionPattern

tvm.relax.backend.cuda#

The Relax CUDA backend compilation pipeline and other passes.

tvm.relax.backend.rocm#

The Relax ROCm backend compilation pipeline and other passes.

tvm.relax.backend.metal#

The Relax Metal backend compilation pipeline and other passes.

tvm.relax.backend.adreno#

The Relax Adreno backend compilation pipeline and other passes.

tvm.relax.backend.gpu_generic#

The Relax Metal backend compilation pipeline and other passes.

tvm.relax.backend.cpu_generic#

The Relax CPU backend compilation pipeline and other passes.

tvm.relax.backend.contrib#

Relax backends contrib