• Lei Wang's avatar
    [Enhancement] Introduce padding annotation and improve memory access validation (#511) · f23c4d30
    Lei Wang authored
    * Added a new attribute `kPaddingMap` in `builtin.h` for managing padding annotations.
    * Enhanced `SafeMemorysRewriter` to utilize an annotated padding map for buffer stores, improving memory access safety.
    * Implemented checks in `layout_inference.cc` to ensure buffers are correctly referenced during layout mapping.
    * Introduced a new test file for validating the padding annotation functionality in TileLang.
    f23c4d30
legalize_safe_memory_access.cc 11.7 KB