void i915_driver_remove(struct drm_i915_private *i915)
 {
-       disable_rpm_wakeref_asserts(&i915->runtime_pm);
+       intel_wakeref_t wakeref;
+
+       wakeref = intel_runtime_pm_get(&i915->runtime_pm);
 
        i915_driver_unregister(i915);
 
 
        i915_driver_hw_remove(i915);
 
-       enable_rpm_wakeref_asserts(&i915->runtime_pm);
+       intel_runtime_pm_put(&i915->runtime_pm, wakeref);
 }
 
 static void i915_driver_release(struct drm_device *dev)
 {
        struct drm_i915_private *dev_priv = to_i915(dev);
        struct intel_runtime_pm *rpm = &dev_priv->runtime_pm;
+       intel_wakeref_t wakeref;
 
        if (!dev_priv->do_release)
                return;
 
-       disable_rpm_wakeref_asserts(rpm);
+       wakeref = intel_runtime_pm_get(rpm);
 
        i915_gem_driver_release(dev_priv);
 
 
        i915_driver_mmio_release(dev_priv);
 
-       enable_rpm_wakeref_asserts(rpm);
+       intel_runtime_pm_put(rpm, wakeref);
+
        intel_runtime_pm_driver_release(rpm);
 
        i915_driver_late_release(dev_priv);