Thermal shutdown handling in data storage device

Methods and apparatus for thermal management in data storage devices are provided. A data storage device includes a first non-volatile memory (NVM), and one or more processors coupled to the first NVM. The one or more processors are configured to receive, from a host, a host command to be performed in connection with the first NVM, the host command comprising a starting logical block address (LBA) and a transfer length, store the starting LBA and the transfer length in a second NVM, attempt to perform the host command, detect, before completion of the performance of the host command, a thermal shutdown state of the first NVM, and invalidate, using the stored starting LBA and the stored transfer length, residual data stored in the first NVM in association with partial completion of the host command prior to the thermal shutdown state.