• Lei Wang's avatar
    [Enhancement] Add strict layout map for improved buffer layout inference (#594) · 18889821
    Lei Wang authored
    - Introduced a `strict_layout_map` to enhance layout inference by ensuring that buffers with strict layout requirements are properly accounted for during the inference process.
    - Updated the inference logic to check for the presence of buffers in the `strict_layout_map` before applying layout changes, improving the accuracy of layout assignments.
    - Refactored the layout inference steps to include the copying of layouts into the new strict map, ensuring a clear separation of layout handling based on inference levels.
    18889821
layout_inference.cc 24 KB