tilelang.tileop.gemm_sp.registry ================================ .. py:module:: tilelang.tileop.gemm_sp.registry Attributes ---------- .. autoapisummary:: tilelang.tileop.gemm_sp.registry.GemmSPTargetPredicate Classes ------- .. autoapisummary:: tilelang.tileop.gemm_sp.registry.GemmSPImplEntry Functions --------- .. autoapisummary:: tilelang.tileop.gemm_sp.registry.register_gemm_sp_impl tilelang.tileop.gemm_sp.registry.resolve_gemm_sp_impl Module Contents --------------- .. py:data:: GemmSPTargetPredicate .. py:class:: GemmSPImplEntry .. py:attribute:: name :type: str .. py:attribute:: predicate :type: GemmSPTargetPredicate .. py:attribute:: impl_class :type: type .. py:function:: register_gemm_sp_impl(name, predicate, impl_class) Register a backend-specific GEMM_SP Python implementation class. .. py:function:: resolve_gemm_sp_impl(target) Resolve the registered GEMM_SP implementation class for a target.