1 files changed, 28 insertions, 1 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
index 4bf9bd96a46e..3751d262de68 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
@@ -46,6 +46,7 @@
 #endif
 #include "vi.h"
 #include "bif/bif_4_1_d.h"
+#include <linux/pci.h>
 
 static int amdgpu_debugfs_regs_init(struct amdgpu_device *adev);
 static void amdgpu_debugfs_regs_cleanup(struct amdgpu_device *adev);
@@ -1181,11 +1182,37 @@ int amdgpu_ip_block_version_cmp(struct amdgpu_device *adev,
         return 1;
 }
 
+static void amdgpu_whether_enable_virtual_display(struct amdgpu_device *adev)
+{
+        adev->enable_virtual_display = false;
+
+        if (amdgpu_virtual_display) {
+                struct drm_device *ddev = adev->ddev;
+                const char *pci_address_name = pci_name(ddev->pdev);
+                char *pciaddstr, *pciaddstr_tmp, *pciaddname;
+
+                pciaddstr = kstrdup(amdgpu_virtual_display, GFP_KERNEL);
+                pciaddstr_tmp = pciaddstr;
+                while ((pciaddname = strsep(&pciaddstr_tmp, ";"))) {
+                        if (!strcmp(pci_address_name, pciaddname)) {
+                                adev->enable_virtual_display = true;
+                                break;
+                        }
+                }
+
+                DRM_INFO("virtual display string:%s, %s:virtual_display:%d\n",
+                                 amdgpu_virtual_display, pci_address_name,
+                                 adev->enable_virtual_display);
+
+                kfree(pciaddstr);
+        }
+}
+
 static int amdgpu_early_init(struct amdgpu_device *adev)
 {
         int i, r;
 
-        DRM_INFO("virtual display enabled:%d\n", amdgpu_virtual_display);
+        amdgpu_whether_enable_virtual_display(adev);
 
         switch (adev->asic_type) {
         case CHIP_TOPAZ: