• Lei Wang's avatar
    [Refactor] Improve layout equality checks and error messaging (#471) · c2480907
    Lei Wang authored
    * [Refactor] Simplify buffer_region_to_tile_region function in copy.py
    
    * Removed redundant logic for handling region extents in the buffer_region_to_tile_region function, streamlining the code for better readability and maintainability.
    * Enhanced error handling by focusing on essential checks while eliminating unnecessary complexity related to variable extents.
    
    * [Refactor] Improve layout equality checks and error messaging
    
    * Updated the `IsEqual` method in `FragmentNode` to ensure consistent evaluation of thread ranges.
    * Enhanced error messaging in `ParallelOp::InferLayout` to include source buffer information for better debugging.
    * Adjusted `ReduceOp::InferLayout` to set thread range during layout condensation, improving layout inference accuracy.
    
    * lintfix
    
    * [Refactor] Rename SetThreadRange to BindThreadRange for clarity
    
    * Updated the `SetThreadRange` method in `FragmentNode` and related classes to `BindThreadRange`, improving method naming consistency and clarity.
    * Adjusted all references to the renamed method across the codebase, ensuring proper functionality and maintaining existing behavior.
    * Enhanced layout equality checks to handle thread ranges more robustly in `IsEqual` method.
    * Updated layout inference methods in `Gemm`, `ParallelOp`, and `ReduceOp` to utilize the new method name, ensuring seamless integration with the updated API.
    
    * [Refactor] Update BindThreadRange usage across layout inference methods
    
    * Modified the implementation of `BindThreadRange` in `FragmentNode` to create a new object instance, enhancing thread range binding functionality.
    * Updated all references to `BindThreadRange` in layout inference methods across `Gemm`, `ParallelOp`, and `ReduceOp` to ensure consistency with the new implementation.
    * Adjusted the return statements in various layout inference functions to utilize the updated method, maintaining existing behavior while improving clarity.
    
    * lint fix
    c2480907
reduce.cc 11.6 KB