tilelang.tools.plot_layout module#

tilelang.tools.plot_layout.plot_layout(layout: Layout, save_directory='./tmp', name: str = 'layout', colormap: str = 'RdPu', verbose: bool = False) → None#

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).

Return type:

None