• Lei Wang's avatar
    [Enhancement] Refactor inflight computing to support dynamic pipeline extents (#1399) · f2858fa1
    Lei Wang authored
    * [Build] Update CMake configuration for tilelang_cython_wrapper installation
    
    - Adjusted output directories for the tilelang_cython_wrapper to ensure that development builds place the extension in build/lib.
    - Updated installation paths to place the extension in tilelang/lib within the wheel, improving organization and avoiding potential conflicts with other modules.
    - Modified the internal library path exposure in env.py to prevent shadowing of common module names, enhancing compatibility and usability in user projects.
    
    * [Build] Standardize output directories for tilelang libraries
    
    - Set output directories for both tilelang and tilelang_module libraries to "${CMAKE_BINARY_DIR}/lib" for consistency in development builds.
    - This change enhances organization and ensures that all build artifacts are located in a unified directory structure.
    
    * [Refactor] Update TVM subproject and enhance pipeline loop handling
    
    - Updated the TVM subproject to commit 90581fe9e5287bbcf1844ad14255a1e1e8cdf7f0.
    - Added new fields to `PipelineAnnotation` and `RewrittenBlockInfo` structures to track original statement indices and improve async state management.
    - Refactored `EmitImpl` and `PopulateWaitCounts` methods to enhance clarity and functionality, including better handling of commit groups and wait counts.
    - Simplified access index calculations and strengthened analyzer constraints for loop bounds.
    
    * [Cleanup] Remove license block and unused includes from inject_pipeline.cc
    
    - Eliminated the Apache license block from the top of the file to streamline the code.
    - Removed unused include directives for memory and stringstream to enhance code clarity and reduce unnecessary dependencies.
    
    * [Refactor] Enhance transformation pipeline and test execution
    
    - Added an additional Simplify transformation in the InjectSoftwarePipeline to improve optimization.
    - Updated the test file to call `test_trival_pipeline()` directly, commenting out the previous main execution for better test isolation.
    f2858fa1
inject_pipeline.cc 44.7 KB