• Yu Cheng's avatar
    [Refactor] Add SetMaxNRegCollector to Improve Register Hint Handling in Warp... · 94c758ad
    Yu Cheng authored
    [Refactor] Add SetMaxNRegCollector to Improve Register Hint Handling in Warp Specialized Rewriter (#194)
    
    * [Refactor] Add SetMaxNRegCollector to Improve Register Hint Handling in Warp Specialized Rewriter
    
    - Introduce `SetMaxNRegCollector` to collect register hints from SetMaxNReg calls
    - Modify `WarpSpecializedRewriter` to use collected register hints for producer and consumer code
    - Add validation checks for register hint values in the collector
    - Remove SetMaxNReg calls during code transformation
    - Enhance flexibility of register allocation in warp specialized rewriting
    
    * temporary remove check in lower_hopper_intrin
    94c758ad
warp_specialized_rewriter.cc 37.4 KB