1 files changed, 5 insertions, 6 deletions

diff --git a/arch/powerpc/platforms/cell/iommu.c b/arch/powerpc/platforms/cell/iommu.c
index c7c8720aa39f..21b502398bf3 100644
--- a/arch/powerpc/platforms/cell/iommu.c
+++ b/arch/powerpc/platforms/cell/iommu.c
@@ -39,6 +39,7 @@
 #include <asm/firmware.h>
 #include <asm/cell-regs.h>
 
+#include "cell.h"
 #include "interrupt.h"
 
 /* Define CELL_IOMMU_REAL_UNMAP to actually unmap non-used pages
@@ -197,7 +198,7 @@ static int tce_build_cell(struct iommu_table *tbl, long index, long npages,
 
         io_pte = (unsigned long *)tbl->it_base + (index - tbl->it_offset);
 
-        for (i = 0; i < npages; i++, uaddr += tbl->it_page_shift)
+        for (i = 0; i < npages; i++, uaddr += (1 << tbl->it_page_shift))
                 io_pte[i] = base_pte | (__pa(uaddr) & CBE_IOPTE_RPN_Mask);
 
         mb();
@@ -857,7 +858,7 @@ static int __init cell_iommu_init_disabled(void)
         cell_dma_direct_offset += base;
 
         if (cell_dma_direct_offset != 0)
-                ppc_md.pci_dma_dev_setup = cell_pci_dma_dev_setup;
+                cell_pci_controller_ops.dma_dev_setup = cell_pci_dma_dev_setup;
 
         printk("iommu: disabled, direct DMA offset is 0x%lx\n",
                cell_dma_direct_offset);
@@ -1197,8 +1198,8 @@ static int __init cell_iommu_init(void)
                 if (cell_iommu_init_disabled() == 0)
                         goto bail;
 
-        /* Setup various ppc_md. callbacks */
-        ppc_md.pci_dma_dev_setup = cell_pci_dma_dev_setup;
+        /* Setup various callbacks */
+        cell_pci_controller_ops.dma_dev_setup = cell_pci_dma_dev_setup;
         ppc_md.dma_get_required_mask = cell_dma_get_required_mask;
         ppc_md.tce_build = tce_build_cell;
         ppc_md.tce_free = tce_free_cell;
@@ -1234,5 +1235,3 @@ static int __init cell_iommu_init(void)
         return 0;
 }
 machine_arch_initcall(cell, cell_iommu_init);
-machine_arch_initcall(celleb_native, cell_iommu_init);
-