#include <linux/init.h>
 #include <linux/interconnect.h>
 #include <linux/ioctl.h>
+#include <linux/delay.h>
 #include <linux/list.h>
 #include <linux/module.h>
 #include <linux/of_device.h>
        mutex_unlock(&core->lock);
 
        disable_irq_nosync(core->irq);
-
-       /*
-        * Delay recovery to ensure venus has completed any pending cache
-        * operations. Without this sleep, we see device reset when firmware is
-        * unloaded after a system error.
-        */
-       schedule_delayed_work(&core->work, msecs_to_jiffies(100));
+       schedule_delayed_work(&core->work, msecs_to_jiffies(10));
 }
 
 static const struct hfi_core_ops venus_core_ops = {
                        container_of(work, struct venus_core, work.work);
        int ret = 0;
 
-       dev_warn(core->dev, "system error has occurred, starting recovery!\n");
-
        pm_runtime_get_sync(core->dev);
 
        hfi_core_deinit(core, true);
-       hfi_destroy(core);
+
+       dev_warn(core->dev, "system error has occurred, starting recovery!\n");
+
        mutex_lock(&core->lock);
+
+       while (pm_runtime_active(core->dev_dec) || pm_runtime_active(core->dev_enc))
+               msleep(10);
+
        venus_shutdown(core);
 
        pm_runtime_put_sync(core->dev);
 
+       while (core->pmdomains[0] && pm_runtime_active(core->pmdomains[0]))
+               usleep_range(1000, 1500);
+
+       hfi_destroy(core);
        ret |= hfi_create(core, &venus_core_ops);
 
        pm_runtime_get_sync(core->dev);
 
        ret |= venus_boot(core);
-
        ret |= hfi_core_resume(core, true);
 
        enable_irq(core->irq);
 
 
        venus_set_state(hdev, VENUS_STATE_DEINIT);
 
-       /*
-        * Once SYS_ERROR received from HW, it is safe to halt the AXI.
-        * With SYS_ERROR, Venus FW may have crashed and HW might be
-        * active and causing unnecessary transactions. Hence it is
-        * safe to stop all AXI transactions from venus subsystem.
-        */
-       venus_halt_axi(hdev);
        venus_sfr_print(hdev);
 }
 
        res = hdev->core->res;
        pkt = hdev->pkt_buf;
 
-       if (hdev->irq_status & WRAPPER_INTR_STATUS_A2HWD_MASK) {
-               venus_sfr_print(hdev);
-               hfi_process_watchdog_timeout(core);
-       }
 
        while (!venus_iface_msgq_read(hdev, pkt)) {
                msg_ret = hfi_process_msg_packet(core, pkt);