drm/amdgpu:changes of virtualization cases probe (v3)

[cascardo/linux.git] / drivers / gpu / drm / amd / amdgpu / amdgpu_drv.c

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c
index c96ae10..0c5f36d 100644 (file)
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c
@@ -485,7 +485,7 @@ amdgpu_pci_shutdown(struct pci_dev *pdev)
        /* if we are running in a VM, make sure the device
         * torn down properly on reboot/shutdown
         */
-       if (adev->virtualization.is_virtual)
+       if (amdgpu_passthrough(adev))
                amdgpu_pci_remove(pdev);
 }