intel_wakeref_t wakeref;
        int ret = 0;
 
+       if (!HAS_HECI_PXP(i915)) {
+               pxp->dev_link = device_link_add(i915_kdev, tee_kdev, DL_FLAG_STATELESS);
+               if (drm_WARN_ON(&i915->drm, !pxp->dev_link))
+                       return -ENODEV;
+       }
+
        mutex_lock(&pxp->tee_mutex);
        pxp->pxp_component = data;
        pxp->pxp_component->tee_dev = tee_kdev;
        mutex_lock(&pxp->tee_mutex);
        pxp->pxp_component = NULL;
        mutex_unlock(&pxp->tee_mutex);
+
+       if (pxp->dev_link) {
+               device_link_del(pxp->dev_link);
+               pxp->dev_link = NULL;
+       }
 }
 
 static const struct component_ops i915_pxp_tee_component_ops = {
 
         * which are protected by &tee_mutex.
         */
        struct i915_pxp_component *pxp_component;
+
+       /* @dev_link: Enforce module relationship for power management ordering. */
+       struct device_link *dev_link;
        /**
         * @pxp_component_added: track if the pxp component has been added.
         * Set and cleared in tee init and fini functions respectively.