• Lei Wang's avatar
    [Enhancement] Improve handling of negative indices for ramp and broadcast node (#1207) · 918a21bd
    Lei Wang authored
    * [Enhancement] Improve handling of negative indices in legalize_negative_index pass
    
    * Added logic to handle scalar and vector indices separately, enhancing the ability to determine non-negativity and negativity of indices.
    * Introduced detailed logging for cases where non-negativity cannot be proven, improving debugging capabilities.
    * Refactored index state determination for vector types, including support for Ramp and Broadcast nodes.
    
    * Fix incorrect lane handling in legalize_negative_index pass by dereferencing lanes to obtain the correct integer value.
    
    * Enhance legalize_negative_index pass by including necessary header for TIR operations. This addition supports improved functionality and maintainability of the transformation logic.
    918a21bd
legalize_negative_index.cc 7.22 KB