1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/infiniband/hw/amso1100/c2.c b/drivers/infiniband/hw/amso1100/c2.c
index 7d79aa361e26..0cfbb6d2f762 100644
--- a/drivers/infiniband/hw/amso1100/c2.c
+++ b/drivers/infiniband/hw/amso1100/c2.c
@@ -989,13 +989,13 @@ static int __devinit c2_probe(struct pci_dev *pcidev,
         }
 
         if ((sizeof(dma_addr_t) > 4)) {
-                ret = pci_set_dma_mask(pcidev, DMA_64BIT_MASK);
+                ret = pci_set_dma_mask(pcidev, DMA_BIT_MASK(64));
                 if (ret < 0) {
                         printk(KERN_ERR PFX "64b DMA configuration failed\n");
                         goto bail2;
                 }
         } else {
-                ret = pci_set_dma_mask(pcidev, DMA_32BIT_MASK);
+                ret = pci_set_dma_mask(pcidev, DMA_BIT_MASK(32));
                 if (ret < 0) {
                         printk(KERN_ERR PFX "32b DMA configuration failed\n");
                         goto bail2;