• Lei Wang's avatar
    [Language] Introduce `T.any_of` and `T.all_of` to reduce a bool arrary (#371) · c4638d65
    Lei Wang authored
    
    
    * [Enhancement] Introduce logical operations `any_of` and `all_of` for buffer checks
    
    - Added new logical operations `any_of` and `all_of` to the TileLang language interface, allowing users to check conditions across buffer elements.
    - Implemented corresponding intrinsic calls for CUDA, enhancing the functionality of the TileLang framework.
    - Updated the `allocate.py` to handle boolean types correctly in shared memory allocations.
    - Introduced tests for the new logical operations to ensure correctness and performance.
    Co-authored-by: default avatarZhiwen Mo <zhiwen.mo25@ic.ac.uk>
    
    * lint fix
    
    ---------
    Co-authored-by: default avatarZhiwen Mo <zhiwen.mo25@ic.ac.uk>
    c4638d65
common.h 5.65 KB