• Lei Wang's avatar
    [Feature] Implement ParallelLoopTransformer for enhanced loop analysis (#295) · 5c8de061
    Lei Wang authored
    * [Feature] Implement ParallelLoopTransformer for enhanced loop analysis
    
    - Introduced the ParallelLoopTransformer class to improve the handling of parallel loops in layout inference.
    - Enhanced the analysis of loop variables and their extents, allowing for more accurate index range calculations.
    - Added a BufferAccessCollector to gather buffer access information, ensuring correct index mapping and condition handling.
    - Updated the LayoutInference pass to utilize the new transformer, improving overall performance and accuracy in loop transformations.
    
    * test fix
    
    * Fix typo in buffer variable documentation and enhance loop variable handling in layout inference. Added checks for related loop variables and improved condition handling for index mapping.
    
    * Refactor loop variable handling in layout inference. Updated loop index variable from `i` to `j` for clarity and improved condition handling for index mapping by replacing `indices[i]` with `index` in predicate construction.
    5c8de061
layout_inference.cc 20.2 KB