Worked on the ROCm/rocMLIR repository to address reliability and correctness in rewrite patterns for convolution and GEMM operations. Focused on resolving a critical use-after-free memory bug in the SortDimensionsMemoryLayout by ensuring essential attributes were preserved before erasing operations, thereby improving memory safety. Updated associated tests to tolerate potential uninitialized memory matches, establishing a minimum correctness threshold and enhancing test stability across configurations. Adjusted MLIR validation checks to reduce flakiness from uninitialized memory, accepting results above a defined threshold. Utilized C++ and MLIR, applying expertise in memory management and performance optimization to deliver robust, maintainable improvements within the codebase.