• Lei Wang's avatar
    [Bugfix][WS] Consider loop min extent when computing phase id (#754) · b39aaf5b
    Lei Wang authored
    * Update test parameters and remove debug print statement
    
    - Adjusted test cases in `test_tilelang_dynamic_symbolic_bench.py` to use smaller matrix sizes (1024x1024) for improved performance and quicker execution.
    - Removed a debug print statement from `phase.py` to clean up the code and enhance clarity.
    
    * Refactor loop stack management in warp_specialized_rewriter
    
    - Introduced a new `LoopInfo` struct to encapsulate loop variable details, including `loop_var`, `extent`, and `min`, enhancing clarity and maintainability.
    - Updated the `loop_stack_` to utilize `LoopInfo` instead of a pair, improving type safety and readability.
    - Adjusted linear index calculations to account for the new structure, ensuring correct behavior in loop transformations.
    b39aaf5b
warp_specialized_rewriter.cc 50 KB