• Yu Cheng's avatar
    [Feature] Introduce NoSetMaxNReg for warp specialization (#289) · 76435ca8
    Yu Cheng authored
    - Added NoSetMaxNReg as a new TIR built-in to indicate no register hint for warp-specialized branches.
    - Updated the warp specialization rewriter to handle the new NoSetMaxNReg operation, allowing for improved register management.
    - Enhanced the Python interface to include NoSetMaxNReg for consistency with TIR operations.
    76435ca8
builtin.cc 4.1 KB