tilelang.backend.gemm_sp

Attributes

GemmSPTargetPredicate

Classes

GemmSPImplEntry

Functions

register_gemm_sp_impl(name, predicate, impl_class)	Register a backend-specific GEMM_SP Python implementation class.
resolve_gemm_sp_impl(target)	Resolve the registered GEMM_SP implementation class for a target.

Module Contents

tilelang.backend.gemm_sp.GemmSPTargetPredicate

class tilelang.backend.gemm_sp.GemmSPImplEntry

name: str

predicate: GemmSPTargetPredicate

impl_class: type

tilelang.backend.gemm_sp.register_gemm_sp_impl(name, predicate, impl_class)

Register a backend-specific GEMM_SP Python implementation class.

Parameters:

• name (str)

• predicate (GemmSPTargetPredicate)

• impl_class (type)

Return type:

None

tilelang.backend.gemm_sp.resolve_gemm_sp_impl(target)

Resolve the registered GEMM_SP implementation class for a target.

Parameters:

target (tvm.target.Target)

Return type:

type