tilelang.language.annotations¶

Annotation helpers exposed on the TileLang language surface.

Functions¶

use_swizzle(panel_size[, order, enable])

Annotate a kernel to use a specific threadblock swizzle pattern.

annotate_layout(layout_map)

Annotate the layout of the buffer.

annotate_safe_value(safe_value_map)

Annotate the safe value of the buffer.

annotate_l2_hit_ratio(l2_hit_ratio_map)

Annotate the L2 hit ratio of the buffer.

Module Contents¶

tilelang.language.annotations.use_swizzle(panel_size, order='row', enable=True)¶

Annotate a kernel to use a specific threadblock swizzle pattern.

Parameters:
  • panel_size (int)

  • order (str)

  • enable (bool)

tilelang.language.annotations.annotate_layout(layout_map)¶

Annotate the layout of the buffer.

Parameters:

layout_map (Dict)

tilelang.language.annotations.annotate_safe_value(safe_value_map)¶

Annotate the safe value of the buffer.

Parameters:

safe_value_map (Dict)

tilelang.language.annotations.annotate_l2_hit_ratio(l2_hit_ratio_map)¶

Annotate the L2 hit ratio of the buffer.

Parameters:

l2_hit_ratio_map (Dict)