#include "tests/xe_test.h"
 #include "xe_bo.h"
 #include "xe_device.h"
+#include "xe_pm.h"
 #include "xe_ttm_vram_mgr.h"
 #include "xe_vm.h"
 
        if (!attach->peer2peer && !xe_bo_can_migrate(gem_to_xe_bo(obj), XE_PL_TT))
                return -EOPNOTSUPP;
 
-       xe_device_mem_access_get(to_xe_device(obj->dev));
+       xe_pm_runtime_get(to_xe_device(obj->dev));
        return 0;
 }
 
 {
        struct drm_gem_object *obj = attach->dmabuf->priv;
 
-       xe_device_mem_access_put(to_xe_device(obj->dev));
+       xe_pm_runtime_put(to_xe_device(obj->dev));
 }
 
 static int xe_dma_buf_pin(struct dma_buf_attachment *attach)