Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Yu Cheng's avatar
    [Bugfix] Add TMA and Producer Buffer Analysis in Warp Specialized Rewriter (#269) · 2abd6ab7
    Yu Cheng authored 
    - Introduced TMAFinder and ProducerUsedBufferFinder classes to analyze TMA loads and identify buffers used in producer conditions.
- Enhanced WarpSpecializedRoleMarker to prepare and utilize the identified buffers during role marking.
- Updated VisitStmt methods to incorporate new analysis logic for IfThenElse and For nodes, improving the handling of TMA loads in the warp specialization process.
    2abd6ab7
warp_specialized_rewriter.cc 40.7 KB