tilelang.carver.template.general_reduce ======================================= .. py:module:: tilelang.carver.template.general_reduce Classes ------- .. autoapisummary:: tilelang.carver.template.general_reduce.GeneralReductionTemplate Module Contents --------------- .. py:class:: GeneralReductionTemplate Bases: :py:obj:`tilelang.carver.template.base.BaseTemplate` .. py:attribute:: structure :type: Union[str, List[str]] :value: None .. py:attribute:: shape :type: List[int] :value: None .. py:attribute:: dtype :type: str :value: 'float16' .. py:method:: get_hardware_aware_configs(arch = None, topk = 10) .. py:method:: initialize_function() Parse the structure (e.g., 'SSR'), build the TVM compute definition with the appropriate spatial and reduce axes, and store it in self._func. .. py:method:: params_as_dict() .. py:property:: class_attributes .. py:method:: __repr__()