Any kind of device memory access should first ensure the device is not
suspended, mmio included.
Signed-off-by: Matthew Auld <matthew.auld@intel.com>
Cc: Rodrigo Vivi <rodrigo.vivi@intel.com>
Reviewed-by: Matthew Brost <matthew.brost@intel.com>
Signed-off-by: Rodrigo Vivi <rodrigo.vivi@intel.com>
         */
        reg = XE_REG(args->addr);
 
+       xe_device_mem_access_get(xe);
        xe_force_wake_get(gt_to_fw(gt), XE_FORCEWAKE_ALL);
 
        if (args->flags & DRM_XE_MMIO_WRITE) {
 
 exit:
        xe_force_wake_put(gt_to_fw(gt), XE_FORCEWAKE_ALL);
+       xe_device_mem_access_put(xe);
 
        return ret;
 }