Worked on the FlagOpen/FlagGems repository to enhance memory safety and kernel reliability in the Ascend backend, focusing on batch matrix multiplication operations. Addressed a critical bug by implementing safe masking for M and N boundary loads during data loading, which prevents out-of-bounds memory access and reduces the risk of runtime failures. Refactored the masking logic to reuse computed masks throughout the data-loading path, improving maintainability and reducing code duplication. This work leveraged Python and Triton, with an emphasis on backend development and memory management, resulting in a more robust and reliable kernel for large-scale production environments.