tilelang.carver.roller.hint module

Hint definition for schedule

class tilelang.carver.roller.hint.Hint

Bases: object

Central configuration class for managing various parameters of computational tasks.

complete_config(node: PrimFuncNode)

classmethod from_dict(dic: Dict) → Hint

property raxis_order: List[int]

property step: List[int]

tensorcore_legalization()

to_dict() → Dict

class tilelang.carver.roller.hint.IntrinInfo(in_dtype: str, out_dtype: str, trans_b: bool, input_transform_kind: int = 0, weight_transform_kind: int = 0)

Bases: object

The information of tensorcore intrinsic related information

property inter_transform_a: bool

property inter_transform_b: bool

is_input_8bit() → bool

property smooth_a: bool

property smooth_b: bool

class tilelang.carver.roller.hint.Stride(stride: int = 1, ax: int = -1)

Bases: object

Manages stride information for a given axis of a tensor.

property ax: int

compute_elements_from_shape(shape: List[int]) → int

compute_strides_from_shape(shape: List[int]) → List[int]

is_valid() → bool

property stride: int

class tilelang.carver.roller.hint.TensorCoreExtraConfig(AS_shape: Tuple[int], BS_shape: Tuple[int], AF_shape: Tuple[int], BF_shape: Tuple[int], tc_axis: Tuple[int])

Bases: object

This class is used to store extra information for tensorcore

class tilelang.carver.roller.hint.TileDict(output_tile)

Bases: object

Manages tiling information and configurations for computational tasks.

get_rstep(node) → Dict[str, int]

get_tile(func) → List[int]