tilelang.backend.gemm

Attributes

GemmTargetPredicate

Classes

GemmImplEntry

Functions

register_gemm_impl(name, inst_name, predicate, impl_class)	Register a backend-specific GEMM implementation class.
resolve_gemm_impl(gemm_inst, target)	Resolve the registered implementation class for a GEMM instruction key.

Module Contents

tilelang.backend.gemm.GemmTargetPredicate

class tilelang.backend.gemm.GemmImplEntry

name: str

inst_name: str

predicate: GemmTargetPredicate

impl_class: type

tilelang.backend.gemm.register_gemm_impl(name, inst_name, predicate, impl_class)

Register a backend-specific GEMM implementation class.

Parameters:

• name (str)

• inst_name (str)

• predicate (GemmTargetPredicate)

• impl_class (type)

Return type:

None

tilelang.backend.gemm.resolve_gemm_impl(gemm_inst, target)

Resolve the registered implementation class for a GEMM instruction key.

Parameters:

• gemm_inst (str)

• target (tvm.target.Target)

Return type:

type