• Yu Cheng's avatar
    [Enhancement] Add warp specialization attribute handling in IR and rewriter (#518) · 41bc15cb
    Yu Cheng authored
    * Introduced an `AttrFrame` for warp specialization in the IR, enhancing the handling of warp-specific optimizations.
    * Refactored the `VisitStmt_` method in `warp_specialized_rewriter.cc` to check for the new warp specialization attribute, improving the detection of warp specialization conditions.
    * Removed outdated code related to condition checks in `IfThenElseNode`, streamlining the specialization logic.
    41bc15cb
warp_specialized_rewriter.cc 40.1 KB