• Lei Wang's avatar
    [Enhancement] Support Auto Layout Inference and Parallelism with variable constraint (#417) · 73a6cb8b
    Lei Wang authored
    * [Enhancement] Introduce thread range management in layout and operation handling
    
    * Added `SetThreadRange` method to `FragmentNode` for managing thread ranges.
    * Updated `LayoutNode::Inverse` to provide more informative error messages.
    * Refactored layout inference and operation lowering to utilize `thread_bounds` instead of `block_size`, enhancing flexibility for thread management.
    * Introduced new tests for tilelang operations to validate thread range functionality and ensure correctness in parallel execution scenarios.
    
    * lint fix
    
    * [Refactor] Improve thread variable handling in layout inference and operation lowering
    
    * Removed workaround for undefined thread_var in layout inference, ensuring proper handling of thread bounds.
    * Updated logic to define thread bounds based on the presence of thread_var, enhancing robustness in thread management.
    * Refactored thread_var initialization in lower_tile_op to maintain consistency across the codebase.
    
    * [Refactor] Update thread variable handling in layout inference and operation lowering
    
    * Refactored thread variable checks to ensure bounds are only accessed when defined, improving safety and clarity.
    * Initialized thread_var with a default range to prevent undefined behavior.
    * Updated logic in lower_tile_op to align with new thread variable handling, enhancing consistency across the codebase.
    73a6cb8b
layout_inference.cc 22.7 KB