1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/net/ehea/ehea_main.c b/drivers/net/ehea/ehea_main.c
index 958dacbb4974..40c34bfe2cfa 100644
--- a/drivers/net/ehea/ehea_main.c
+++ b/drivers/net/ehea/ehea_main.c
@@ -3517,12 +3517,14 @@ static int ehea_mem_notifier(struct notifier_block *nb,
                 /* Readd canceled memory block */
         case MEM_ONLINE:
                 ehea_info("memory is going online");
+                set_bit(__EHEA_STOP_XFER, &ehea_driver_flags);
                 if (ehea_add_sect_bmap(arg->start_pfn, arg->nr_pages))
                         return NOTIFY_BAD;
                 ehea_rereg_mrs(NULL);
                 break;
         case MEM_GOING_OFFLINE:
                 ehea_info("memory is going offline");
+                set_bit(__EHEA_STOP_XFER, &ehea_driver_flags);
                 if (ehea_rem_sect_bmap(arg->start_pfn, arg->nr_pages))
                         return NOTIFY_BAD;
                 ehea_rereg_mrs(NULL);