static int dra7xx_pcie_msi_host_init(struct pcie_port *pp)
 {
        struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
+       struct device *dev = pci->dev;
        u32 ctrl, num_ctrls;
+       int ret;
 
        pp->msi_irq_chip = &dra7xx_pci_msi_bottom_irq_chip;
 
                                    ~0);
        }
 
-       return dw_pcie_allocate_domains(pp);
+       ret = dw_pcie_allocate_domains(pp);
+       if (ret)
+               return ret;
+
+       pp->msi_data = dma_map_single_attrs(dev, &pp->msi_msg,
+                                          sizeof(pp->msi_msg),
+                                          DMA_FROM_DEVICE,
+                                          DMA_ATTR_SKIP_CPU_SYNC);
+       ret = dma_mapping_error(dev, pp->msi_data);
+       if (ret) {
+               dev_err(dev, "Failed to map MSI data\n");
+               pp->msi_data = 0;
+               dw_pcie_free_msi(pp);
+       }
+       return ret;
 }
 
 static const struct dw_pcie_host_ops dra7xx_pcie_host_ops = {
 
        irq_domain_remove(pp->msi_domain);
        irq_domain_remove(pp->irq_domain);
 
-       if (pp->msi_page)
-               __free_page(pp->msi_page);
+       if (pp->msi_data) {
+               struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
+               struct device *dev = pci->dev;
+
+               dma_unmap_single_attrs(dev, pp->msi_data, sizeof(pp->msi_msg),
+                                      DMA_FROM_DEVICE, DMA_ATTR_SKIP_CPU_SYNC);
+       }
 }
 
 void dw_pcie_msi_init(struct pcie_port *pp)
 {
        struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
-       struct device *dev = pci->dev;
-       u64 msi_target;
+       u64 msi_target = (u64)pp->msi_data;
 
        if (!IS_ENABLED(CONFIG_PCI_MSI))
                return;
 
-       pp->msi_page = alloc_page(GFP_KERNEL);
-       pp->msi_data = dma_map_page(dev, pp->msi_page, 0, PAGE_SIZE,
-                                   DMA_FROM_DEVICE);
-       if (dma_mapping_error(dev, pp->msi_data)) {
-               dev_err(dev, "Failed to map MSI data\n");
-               __free_page(pp->msi_page);
-               pp->msi_page = NULL;
-               return;
-       }
-       msi_target = (u64)pp->msi_data;
-
        /* Program the msi_data */
        dw_pcie_writel_dbi(pci, PCIE_MSI_ADDR_LO, lower_32_bits(msi_target));
        dw_pcie_writel_dbi(pci, PCIE_MSI_ADDR_HI, upper_32_bits(msi_target));
                                irq_set_chained_handler_and_data(pp->msi_irq,
                                                            dw_chained_msi_isr,
                                                            pp);
+
+                       pp->msi_data = dma_map_single_attrs(pci->dev, &pp->msi_msg,
+                                                     sizeof(pp->msi_msg),
+                                                     DMA_FROM_DEVICE,
+                                                     DMA_ATTR_SKIP_CPU_SYNC);
+                       if (dma_mapping_error(pci->dev, pp->msi_data)) {
+                               dev_err(pci->dev, "Failed to map MSI data\n");
+                               pp->msi_data = 0;
+                               goto err_free_msi;
+                       }
                } else {
                        ret = pp->ops->msi_host_init(pp);
                        if (ret < 0)