• Lei Wang's avatar
    [Layout] Strict annotate completed replicated layout for fragment with constant index (#929) · fc4bd452
    Lei Wang authored
    * [Layout] Add IsCompletedReplicated method and enhance layout inference in ParallelOpNode
    
    - Introduced IsCompletedReplicated method in FragmentNode to check if a buffer is fully replicated.
    - Enhanced InferLayout in ParallelOpNode to handle layout inference for replicated buffers, ensuring only fragment[0] access is allowed.
    - Updated error handling for non-zero index access in fragment buffers to improve robustness.
    
    * [Layout] Improve code formatting and readability in layout.cc and parallel.cc
    
    - Enhanced formatting in FragmentNode's IsCompletedReplicated method for better clarity.
    - Updated InferLayout method in ParallelOpNode to improve code readability by adjusting line breaks and indentation.
    - Ensured consistent formatting across conditional statements and comments for improved maintainability.
    
    * updt
    
    * optimize const index related op
    
    * bug fix
    
    * reduce gdn test
    
    * test fix
    
    * lintfix
    
    * lint fix
    
    * test fix
    fc4bd452
example_gemm_persistent.py 6.11 KB