Worked on the torvalds/linux repository to enhance the robustness and concurrency of TLB invalidation within the Linux kernel, focusing on the DRM/XE driver path. Addressed concurrency challenges by refactoring locking mechanisms and introducing a cleanup routine during driver teardown, which prevents memory leaks and dangling pointers. Implemented proactive shutdown safety by canceling pending TLB invalidation workers, reducing the risk of race conditions during driver removal. These changes improved system stability and reliability in memory-management intensive and graphics workloads. The work was carried out using C programming, with an emphasis on kernel development, concurrency control, and mutex handling for safe resource management.