• Cunxiao Ni's avatar
    [Tool] Provide layout visualization tool (#1353) · 924225ed
    Cunxiao Ni authored
    * Provide layout visualization tool
    
    Adds a layout visualization tool to TileLang, which helps users understand and debug the layout transformations applied during compilation.
    
    This tool visualizes the memory layout of tensors at different stages of the compilation process, allowing developers to identify potential inefficiencies and optimize their code for better performance.
    
    The visualization can be enabled via a pass config option.
    
    * format
    
    * add layout visual example
    
    * Adds vis extra with matplotlib dependency
    
    * rafactor pass config name
    
    * fix lint
    
    * Enables configurable layout visualization formats
    
    Allows users to specify the output formats (png, pdf, svg) for layout visualization through a pass config option.
    
    This change provides more flexibility in how layout visualizations are generated, allowing users to choose the formats that best suit their needs.
    
    It also fixes a bug where layout visualization was not correctly disabled when the config option was set to "false".
    
    * Adds visual layout inference tool docs
    
    * fix lint
    
    * fix lint
    
    * Rafactor configurable layout visualization formats
    
    * fix lint
    
    * fix typo
    
    * add some comments
    
    * fix lints
    
    * add some warnings for user
    
    * Moves layout visualization
    
    * Refactors layout visualization pass configuration
    
    Updates the layout visualization pass configuration to use boolean flag for enabling and a string for specifying formats.
    
    * Enables multiple layout visualization formats
    
    * Updates layout visualization docs
    
    * Moves layout visualization to analysis
    924225ed
builtin.cc 14 KB