[PATCH 1/3] drm/amdgpu: save/restore part of xcp drm_device fields

[James Zhu]

Redirect xcp allocated drm_device::rdev/pdev/driver with
amdgpu pci_device/drm_device setting. They need be saved
before redirect and restored after unregister xcp drm_device.

Signed-off-by: James Zhu <james....@amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.c | 16 ++++++++++++++--
 drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.h |  4 ++++
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.c
index daeb6bcc9245..6ab5713ff741 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.c
@@ -240,9 +240,14 @@ static int amdgpu_xcp_dev_alloc(struct amdgpu_device *adev)
                        return PTR_ERR(p_ddev);
 
                /* Redirect all IOCTLs to the primary device */
+               adev->xcp_mgr->xcp[i].rdev = p_ddev->render->dev;
+               adev->xcp_mgr->xcp[i].pdev = p_ddev->primary->dev;
+               adev->xcp_mgr->xcp[i].driver = p_ddev->driver;
+               adev->xcp_mgr->xcp[i].vma_offset_manager = 
p_ddev->vma_offset_manager;
                p_ddev->render->dev = ddev;
                p_ddev->primary->dev = ddev;
                p_ddev->vma_offset_manager = ddev->vma_offset_manager;
+               p_ddev->driver = &amdgpu_partition_driver;
                adev->xcp_mgr->xcp[i].ddev = p_ddev;
        }
 
@@ -330,13 +335,20 @@ int amdgpu_xcp_dev_register(struct amdgpu_device *adev,
 
 void amdgpu_xcp_dev_unplug(struct amdgpu_device *adev)
 {
+       struct drm_device *p_ddev;
        int i;
 
        if (!adev->xcp_mgr)
                return;
 
-       for (i = 0; i < MAX_XCP; i++)
-               drm_dev_unplug(adev->xcp_mgr->xcp[i].ddev);
+       for (i = 0; i < MAX_XCP; i++) {
+               p_ddev = adev->xcp_mgr->xcp[i].ddev;
+               drm_dev_unplug(p_ddev);
+               p_ddev->render->dev = adev->xcp_mgr->xcp[i].rdev;
+               p_ddev->primary->dev = adev->xcp_mgr->xcp[i].pdev;
+               p_ddev->driver =  adev->xcp_mgr->xcp[i].driver;
+               p_ddev->vma_offset_manager = 
adev->xcp_mgr->xcp[i].vma_offset_manager;
+       }
 }
 
 int amdgpu_xcp_open_device(struct amdgpu_device *adev,
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.h 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.h
index 9c5912b9d8bd..0f8026d64ea5 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_xcp.h
@@ -77,6 +77,10 @@ struct amdgpu_xcp {
        bool valid;
        atomic_t        ref_cnt;
        struct drm_device *ddev;
+       struct drm_device *rdev;
+       struct drm_device *pdev;
+       struct drm_driver *driver;
+       struct drm_vma_offset_manager *vma_offset_manager;
        struct amdgpu_sched     
gpu_sched[AMDGPU_HW_IP_NUM][AMDGPU_RING_PRIO_MAX];
 };
 
-- 
2.34.1