• Lei Wang's avatar
    [Enhancement] Improve iterator handling in layout utilities and parallel operations (#1221) · 2957afca
    Lei Wang authored
    * [Enhancement] Improve iterator handling in layout utilities and parallel operations
    
    * Added a new function, DivideUnusedIterators, to detect per-iterator gaps in fused index expressions, enhancing the accuracy of unused iterator detection.
    * Updated CompleteBufferFragment to prefer direct inversion for bijective index mappings and introduced a fallback mechanism for non-bijective cases, improving layout inversion robustness.
    * Added a new test for layout inference in fused kernels to ensure correct compilation and execution without layout inversion failures.
    
    * lint fix
    2957afca
test_tilelang_layout_fused_replicate.py 1.71 KB