• Lei Wang's avatar
    [Enhancement] Support Layout/Fragment Reshape (#1241) · 4370309b
    Lei Wang authored
    
    
    * Update layout handling and introduce reshape functionality
    
    - Updated the `LayoutNode` class to include a new `Reshape` method, allowing for dynamic reshaping of layouts based on input shapes.
    - Enhanced the `OutputShape` method to provide better handling of cases where the analyzer cannot form an `IntervalSet`, implementing fallback mechanisms to ensure safe extents.
    - Refactored the `ReduceOpNode` to utilize `BufferRegion` for improved memory handling during reduction operations.
    - Added tests for reshaping functionality and layout transformations to ensure correctness and performance in various scenarios.
    
    * lint fix
    
    * Revert tvm submodule pointer to 1815c3e0b6ec4ead36370bbd1562025d8529017c; keep src unchanged
    
    * Update tvm submodule to commit f0bbd3bf741413c35c389ba5dedd5be206000ad1
    
    * Update tvm submodule to commit f0bbd3bf741413c35c389ba5dedd5be206000ad1
    
    * remove useless prove
    
    * remove comment
    
    ---------
    Co-authored-by: default avatartilelang-bot <bot@tilelang>
    4370309b
reduce.cc 21.8 KB