#include <asm/octeon/cvmx-npi-defs.h>
 #include <asm/octeon/cvmx-pci-defs.h>
 #include <asm/octeon/cvmx-npei-defs.h>
+#include <asm/octeon/cvmx-sli-defs.h>
 #include <asm/octeon/cvmx-pexp-defs.h>
 #include <asm/octeon/pci-octeon.h>
 
                msg.address_lo = (0 + CVMX_NPEI_PCIE_MSI_RCV) & 0xffffffff;
                msg.address_hi = (0 + CVMX_NPEI_PCIE_MSI_RCV) >> 32;
                break;
+       case OCTEON_DMA_BAR_TYPE_PCIE2:
+               /* When using PCIe2, Bar 0 is based at 0 */
+               msg.address_lo = (0 + CVMX_SLI_PCIE_MSI_RCV) & 0xffffffff;
+               msg.address_hi = (0 + CVMX_SLI_PCIE_MSI_RCV) >> 32;
+               break;
        default:
                panic("arch_setup_msi_irq: Invalid octeon_dma_bar_type");
        }