tilelang.carver.template.general_reduce

類別

GeneralReductionTemplate

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)

參數:

• arch (tilelang.carver.arch.TileDevice)

• topk (int)

回傳型別:

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.

回傳型別:

None

params_as_dict()

property class_attributes

__repr__()

回傳型別:

str