Data recovery after storage failure in a memory system

A non-transitory machine-readable storage medium storing instructions, which when executed by a processing device, cause the processing device to perform a method comprising:

• receiving a first physical address and a data, wherein the first physical address corresponds to a logical address;
• initiating storage of the data at the first physical address in a first of a plurality of non-volatile memory dice of a memory subsystem by sending the data to the first non-volatile memory die, wherein each of the plurality of non-volatile memory dice includes volatile storage and non-volatile storage, and wherein the first non-volatile memory die temporarily stores the data in the volatile storage of the first non-volatile memory die while writing to the non-volatile storage of the first non-volatile memory die;
• sending a message indicating that the data has been successfully stored in the first non-volatile memory die, wherein the message is sent before determining whether or not the data was successfully stored in the non-volatile storage of the first non-volatile memory die;
• retrieving the data from the volatile storage of the first non-volatile memory die in response to determining that storage of the data at the first physical address in the non-volatile memory die has failed;
• sending a request for a second physical address in response to determining that storage of the data at the first physical address has failed, wherein the request includes the first physical address;
• receiving a second physical address corresponding to the logical address;
• and
• storing the data at the second physical address in one of the plurality of non-volatile memory dice.

A memory subsystem comprising:

• a controller memory;
• a plurality of non-volatile memory dice each including volatile storage and non-volatile storage;
• and
• a processing device coupled to the non-volatile memory die, the processing device configured to:
• receive a first physical address and a data, wherein the first physical address corresponds to a logical address;
• initiate storage of the data at the first physical address in a first of the plurality of non-volatile memory dice of a memory subsystem by sending the data to the first non-volatile memory die, wherein the first non-volatile memory die temporarily stores the data in the volatile storage of the first non-volatile memory die while writing to the non-volatile storage of the first non-volatile memory die;
• send a message indicating that the data has been successfully stored in the first non-volatile memory die, wherein the message is sent before determining whether or not the data was successfully stored in the non-volatile storage of the first non-volatile memory die;
• retrieve the data from the volatile storage of the first non-volatile memory die in response to determining that storage of the data at the first physical address in the non-volatile memory die has failed;
• send a request for a second physical address in response to determining that storage of the data at the first physical address has failed, wherein the request includes the first physical address;
• receive a second physical address corresponding to the logical address;
• and
• initiate storage of the data at the second physical address in one of the plurality of non-volatile memory dice.