tilelang.tools.plot_layout¶

Functions¶

plot_layout(layout[, save_directory, name, colormap, ...])

Plot the layout of a buffer.

tilelang.tools.plot_layout.plot_layout(layout, save_directory='./tmp', name='layout', colormap='RdPu', verbose=False, formats='png')¶

Plot the layout of a buffer.

Parameters:

layout (T.Layout) – The layout object that describes how indices are mapped.
save_directory (str, optional) – The directory where the output images will be saved (default is “./tmp”).
name (str, optional) – The base name of the output files (default is “layout”).
colormap (str, optional) – The colormap to use for visualization (default is “RdPu”).
verbose (bool, optional) – If True, prints additional information about the mapping (default is False).
formats (str | list[str], optional) – The formats to save the image in (default is “png”).

Return type:

None