if (rc)
                goto err_out_disable_device;
 
-       if ((pci_set_dma_mask(pdev, DMA_BIT_MASK(64)) != 0) ||
-           (pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(64)) != 0)) {
-               if ((pci_set_dma_mask(pdev, DMA_BIT_MASK(32)) != 0) ||
-                  (pci_set_consistent_dma_mask(pdev, DMA_BIT_MASK(32)) != 0)) {
-                       dev_err(dev, "No usable DMA addressing method\n");
-                       rc = -EIO;
-                       goto err_out_regions;
-               }
+       if (dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(64)) ||
+           dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(32))) {
+               dev_err(dev, "No usable DMA addressing method\n");
+               rc = -EIO;
+               goto err_out_regions;
        }
 
        shost = hisi_sas_shost_alloc_pci(pdev);