tvm.relax.backend

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