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` Base class template for hardware-aware configurations. This serves as an abstract base class (ABC) that defines the structure for subclasses implementing hardware-specific optimizations. .. 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) Abstract method that must be implemented by subclasses. It should return a list of hardware-aware configurations (hints) based on the specified architecture. :param arch: The target architecture. Defaults to None. :type arch: TileDevice, optional :param topk: Number of top configurations to return. Defaults to 10. :type topk: int, optional :returns: A list of recommended hardware-aware configurations. :rtype: List[Hint] .. 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__()