#include "xe_migrate.h"
 #include "xe_mmio.h"
 #include "xe_pat.h"
+#include "xe_pm.h"
 #include "xe_mocs.h"
 #include "xe_reg_sr.h"
 #include "xe_ring_ops.h"
                goto err_fail;
        }
 
+       xe_pm_runtime_get(gt_to_xe(gt));
        xe_gt_sanitize(gt);
 
-       xe_device_mem_access_get(gt_to_xe(gt));
        err = xe_force_wake_get(gt_to_fw(gt), XE_FORCEWAKE_ALL);
        if (err)
                goto err_msg;
                goto err_out;
 
        err = xe_force_wake_put(gt_to_fw(gt), XE_FORCEWAKE_ALL);
-       xe_device_mem_access_put(gt_to_xe(gt));
        XE_WARN_ON(err);
+       xe_pm_runtime_put(gt_to_xe(gt));
 
        xe_gt_info(gt, "reset done\n");
 
        XE_WARN_ON(xe_force_wake_put(gt_to_fw(gt), XE_FORCEWAKE_ALL));
 err_msg:
        XE_WARN_ON(xe_uc_start(>->uc));
-       xe_device_mem_access_put(gt_to_xe(gt));
+       xe_pm_runtime_put(gt_to_xe(gt));
 err_fail:
        xe_gt_err(gt, "reset failed (%pe)\n", ERR_PTR(err));