1 files changed, 1 insertions, 1 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_drv.c
index c96ae105b764..0c5f36d1ea3e 100644
--- 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);
 }