tilelang.carver.template.general_reduce¶
Classes¶
Module Contents¶
- class tilelang.carver.template.general_reduce.GeneralReductionTemplate¶
Bases:
tilelang.carver.template.base.BaseTemplate- structure: str | List[str] = None¶
- shape: List[int] = None¶
- dtype: str = 'float16'¶
- get_hardware_aware_configs(arch=None, topk=10)¶
- Parameters:
arch (tilelang.carver.arch.TileDevice)
topk (int)
- Return type:
List[tilelang.carver.roller.Hint]
- 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.
- Return type:
None
- params_as_dict()¶
- property class_attributes¶
- __repr__()¶
- Return type:
str