1 files changed, 4 insertions, 6 deletions

diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
index fdcf01af6762..b2d23d1b0d41 100644
--- a/drivers/pci/pci.c
+++ b/drivers/pci/pci.c
@@ -2305,15 +2305,13 @@ void pci_msi_off(struct pci_dev *dev)
 int
 pci_set_dma_mask(struct pci_dev *dev, u64 mask)
 {
-        if (!pci_dma_supported(dev, mask))
-                return -EIO;
-
-        dev->dma_mask = mask;
+        int ret = dma_set_mask(&dev->dev, mask);
+        if (ret)
+                return ret;
         dev_dbg(&dev->dev, "using %dbit DMA mask\n", fls64(mask));
-
         return 0;
 }
-    
+
 int
 pci_set_consistent_dma_mask(struct pci_dev *dev, u64 mask)
 {