Merge branch 'linus' into x86/defconfig

8 files changed, 169 insertions, 203 deletions

diff --git a/arch/powerpc/platforms/cell/spufs/run.c b/arch/powerpc/platforms/cell/spufs/run.c
index f7edba6cb795..c9bb7cfd3dca 100644
--- a/arch/powerpc/platforms/cell/spufs/run.c
+++ b/arch/powerpc/platforms/cell/spufs/run.c
@@ -206,11 +206,6 @@ static int spu_run_init(struct spu_context *ctx, u32 *npc)
                         (SPU_RUNCNTL_RUNNABLE | SPU_RUNCNTL_ISOLATE);
                 if (runcntl == 0)
                         runcntl = SPU_RUNCNTL_RUNNABLE;
-        }
-
-        if (ctx->flags & SPU_CREATE_NOSCHED) {
-                spuctx_switch_state(ctx, SPU_UTIL_USER);
-                ctx->ops->runcntl_write(ctx, runcntl);
         } else {
                 unsigned long privcntl;
 
@@ -219,9 +214,15 @@ static int spu_run_init(struct spu_context *ctx, u32 *npc)
                 else
                         privcntl = SPU_PRIVCNTL_MODE_NORMAL;
 
-                ctx->ops->npc_write(ctx, *npc);
                 ctx->ops->privcntl_write(ctx, privcntl);
-                ctx->ops->runcntl_write(ctx, runcntl);
+                ctx->ops->npc_write(ctx, *npc);
+        }
+
+        ctx->ops->runcntl_write(ctx, runcntl);
+
+        if (ctx->flags & SPU_CREATE_NOSCHED) {
+                spuctx_switch_state(ctx, SPU_UTIL_USER);
+        } else {
 
                 if (ctx->state == SPU_STATE_SAVED) {
                         ret = spu_activate(ctx, 0);
diff --git a/arch/powerpc/platforms/cell/spufs/sched.c b/arch/powerpc/platforms/cell/spufs/sched.c
index 2deeeba7eccf..1c1b627ee843 100644
--- a/arch/powerpc/platforms/cell/spufs/sched.c
+++ b/arch/powerpc/platforms/cell/spufs/sched.c
@@ -641,8 +641,10 @@ static struct spu *find_victim(struct spu_context *ctx)
 
                         if (tmp && tmp->prio > ctx->prio &&
                             !(tmp->flags & SPU_CREATE_NOSCHED) &&
-                            (!victim || tmp->prio > victim->prio))
+                            (!victim || tmp->prio > victim->prio)) {
                                 victim = spu->ctx;
+                                get_spu_context(victim);
+                        }
                 }
                 mutex_unlock(&cbe_spu_info[node].list_mutex);
 
@@ -658,6 +660,7 @@ static struct spu *find_victim(struct spu_context *ctx)
                          * look at another context or give up after X retries.
                          */
                         if (!mutex_trylock(&victim->state_mutex)) {
+                                put_spu_context(victim);
                                 victim = NULL;
                                 goto restart;
                         }
@@ -670,6 +673,7 @@ static struct spu *find_victim(struct spu_context *ctx)
                                  * restart the search.
                                  */
                                 mutex_unlock(&victim->state_mutex);
+                                put_spu_context(victim);
                                 victim = NULL;
                                 goto restart;
                         }
@@ -687,6 +691,7 @@ static struct spu *find_victim(struct spu_context *ctx)
                                 spu_add_to_rq(victim);
 
                         mutex_unlock(&victim->state_mutex);
+                        put_spu_context(victim);
 
                         return spu;
                 }
@@ -985,9 +990,11 @@ static int spusched_thread(void *unused)
                                 struct spu_context *ctx = spu->ctx;
 
                                 if (ctx) {
+                                        get_spu_context(ctx);
                                         mutex_unlock(mtx);
                                         spusched_tick(ctx);
                                         mutex_lock(mtx);
+                                        put_spu_context(ctx);
                                 }
                         }
                         mutex_unlock(mtx);
@@ -1030,7 +1037,7 @@ void spuctx_switch_state(struct spu_context *ctx,
                 node = spu->node;
                 if (old_state == SPU_UTIL_USER)
                         atomic_dec(&cbe_spu_info[node].busy_spus);
-                if (new_state == SPU_UTIL_USER);
+                if (new_state == SPU_UTIL_USER)
                         atomic_inc(&cbe_spu_info[node].busy_spus);
         }
 }
diff --git a/arch/powerpc/platforms/powermac/Makefile b/arch/powerpc/platforms/powermac/Makefile
index 58ecdd72630f..be60d64be7ad 100644
--- a/arch/powerpc/platforms/powermac/Makefile
+++ b/arch/powerpc/platforms/powermac/Makefile
@@ -2,7 +2,7 @@ CFLAGS_bootx_init.o  		+= -fPIC
 
 ifdef CONFIG_FTRACE
 # Do not trace early boot code
-CFLAGS_REMOVE_bootx_init.o = -pg
+CFLAGS_REMOVE_bootx_init.o = -pg -mno-sched-epilog
 endif
 
 obj-y                           += pic.o setup.o time.o feature.o pci.o \
diff --git a/arch/powerpc/platforms/ps3/htab.c b/arch/powerpc/platforms/ps3/htab.c
index 1cf901fa9031..6eb1d4d182c9 100644
--- a/arch/powerpc/platforms/ps3/htab.c
+++ b/arch/powerpc/platforms/ps3/htab.c
@@ -29,138 +29,75 @@
 
 #include "platform.h"
 
-#if defined(DEBUG)
-#define DBG udbg_printf
-#else
-#define DBG pr_debug
-#endif
-
-static struct hash_pte *htab;
-static unsigned long htab_addr;
-static unsigned char *bolttab;
-static unsigned char *inusetab;
-
-static DEFINE_SPINLOCK(ps3_bolttab_lock);
-
-#define debug_dump_hpte(_a, _b, _c, _d, _e, _f, _g) \
-        _debug_dump_hpte(_a, _b, _c, _d, _e, _f, _g, __func__, __LINE__)
-static void _debug_dump_hpte(unsigned long pa, unsigned long va,
-        unsigned long group, unsigned long bitmap, struct hash_pte lhpte,
-        int psize, unsigned long slot, const char* func, int line)
-{
-        DBG("%s:%d: pa     = %lxh\n", func, line, pa);
-        DBG("%s:%d: lpar   = %lxh\n", func, line,
-                ps3_mm_phys_to_lpar(pa));
-        DBG("%s:%d: va     = %lxh\n", func, line, va);
-        DBG("%s:%d: group  = %lxh\n", func, line, group);
-        DBG("%s:%d: bitmap = %lxh\n", func, line, bitmap);
-        DBG("%s:%d: hpte.v = %lxh\n", func, line, lhpte.v);
-        DBG("%s:%d: hpte.r = %lxh\n", func, line, lhpte.r);
-        DBG("%s:%d: psize  = %xh\n", func, line, psize);
-        DBG("%s:%d: slot   = %lxh\n", func, line, slot);
-}
+/**
+ * enum lpar_vas_id - id of LPAR virtual address space.
+ * @lpar_vas_id_current: Current selected virtual address space
+ *
+ * Identify the target LPAR address space.
+ */
+
+enum ps3_lpar_vas_id {
+        PS3_LPAR_VAS_ID_CURRENT = 0,
+};
+
+
+static DEFINE_SPINLOCK(ps3_htab_lock);
 
 static long ps3_hpte_insert(unsigned long hpte_group, unsigned long va,
         unsigned long pa, unsigned long rflags, unsigned long vflags,
         int psize, int ssize)
 {
-        unsigned long slot;
-        struct hash_pte lhpte;
-        int secondary = 0;
-        unsigned long result;
-        unsigned long bitmap;
+        int result;
+        u64 hpte_v, hpte_r;
+        u64 inserted_index;
+        u64 evicted_v, evicted_r;
+        u64 hpte_v_array[4], hpte_rs;
         unsigned long flags;
-        unsigned long p_pteg, s_pteg, b_index, b_mask, cb, ci;
-
-        vflags &= ~HPTE_V_SECONDARY; /* this bit is ignored */
-
-        lhpte.v = hpte_encode_v(va, psize, MMU_SEGSIZE_256M) |
-                vflags | HPTE_V_VALID;
-        lhpte.r = hpte_encode_r(ps3_mm_phys_to_lpar(pa), psize) | rflags;
-
-        p_pteg = hpte_group / HPTES_PER_GROUP;
-        s_pteg = ~p_pteg & htab_hash_mask;
-
-        spin_lock_irqsave(&ps3_bolttab_lock, flags);
-
-        BUG_ON(bolttab[p_pteg] == 0xff && bolttab[s_pteg] == 0xff);
+        long ret = -1;
 
-        bitmap = (inusetab[p_pteg] << 8) | inusetab[s_pteg];
+        /*
+         * lv1_insert_htab_entry() will search for victim
+         * entry in both primary and secondary pte group
+         */
+        vflags &= ~HPTE_V_SECONDARY;
 
-        if (bitmap == 0xffff) {
-                /*
-                 * PTEG is full. Search for victim.
-                 */
-                bitmap &= ~((bolttab[p_pteg] << 8) | bolttab[s_pteg]);
-                do {
-                        ci = mftb() & 15;
-                        cb = 0x8000UL >> ci;
-                } while ((cb & bitmap) == 0);
-        } else {
-                /*
-                 * search free slot in hardware order
-                 *      [primary]       0, 2, 4, 6, 1, 3, 5, 7
-                 *      [secondary]     0, 2, 4, 6, 1, 3, 5, 7
-                 */
-                for (ci = 0; ci < HPTES_PER_GROUP; ci += 2) {
-                        cb = 0x8000UL >> ci;
-                        if ((cb & bitmap) == 0)
-                                goto found;
-                }
-                for (ci = 1; ci < HPTES_PER_GROUP; ci += 2) {
-                        cb = 0x8000UL >> ci;
-                        if ((cb & bitmap) == 0)
-                                goto found;
-                }
-                for (ci = HPTES_PER_GROUP; ci < HPTES_PER_GROUP*2; ci += 2) {
-                        cb = 0x8000UL >> ci;
-                        if ((cb & bitmap) == 0)
-                                goto found;
-                }
-                for (ci = HPTES_PER_GROUP+1; ci < HPTES_PER_GROUP*2; ci += 2) {
-                        cb = 0x8000UL >> ci;
-                        if ((cb & bitmap) == 0)
-                                goto found;
-                }
-        }
+        hpte_v = hpte_encode_v(va, psize, ssize) | vflags | HPTE_V_VALID;
+        hpte_r = hpte_encode_r(ps3_mm_phys_to_lpar(pa), psize) | rflags;
 
-found:
-        if (ci < HPTES_PER_GROUP) {
-                slot = p_pteg * HPTES_PER_GROUP + ci;
-        } else {
-                slot = s_pteg * HPTES_PER_GROUP + (ci & 7);
-                /* lhpte.dw0.dw0.h = 1; */
-                vflags |= HPTE_V_SECONDARY;
-                lhpte.v |= HPTE_V_SECONDARY;
-        }
+        spin_lock_irqsave(&ps3_htab_lock, flags);
 
-        result = lv1_write_htab_entry(0, slot, lhpte.v, lhpte.r);
+        /* talk hvc to replace entries BOLTED == 0 */
+        result = lv1_insert_htab_entry(PS3_LPAR_VAS_ID_CURRENT, hpte_group,
+                                       hpte_v, hpte_r,
+                                       HPTE_V_BOLTED, 0,
+                                       &inserted_index,
+                                       &evicted_v, &evicted_r);
 
         if (result) {
-                debug_dump_hpte(pa, va, hpte_group, bitmap, lhpte, psize, slot);
+                /* all entries bolted !*/
+                pr_info("%s:result=%d va=%lx pa=%lx ix=%lx v=%lx r=%lx\n",
+                        __func__, result, va, pa, hpte_group, hpte_v, hpte_r);
                 BUG();
         }
 
         /*
-         * If used slot is not in primary HPTE group,
-         * the slot should be in secondary HPTE group.
+         * see if the entry is inserted into secondary pteg
          */
+        result = lv1_read_htab_entries(PS3_LPAR_VAS_ID_CURRENT,
+                                       inserted_index & ~0x3UL,
+                                       &hpte_v_array[0], &hpte_v_array[1],
+                                       &hpte_v_array[2], &hpte_v_array[3],
+                                       &hpte_rs);
+        BUG_ON(result);
 
-        if ((hpte_group ^ slot) & ~(HPTES_PER_GROUP - 1)) {
-                secondary = 1;
-                b_index = s_pteg;
-        } else {
-                secondary = 0;
-                b_index = p_pteg;
-        }
+        if (hpte_v_array[inserted_index % 4] & HPTE_V_SECONDARY)
+                ret = (inserted_index & 7) | (1 << 3);
+        else
+                ret = inserted_index & 7;
 
-        b_mask = (lhpte.v & HPTE_V_BOLTED) ? 1 << 7 : 0 << 7;
-        bolttab[b_index] |= b_mask >> (slot & 7);
-        b_mask = 1 << 7;
-        inusetab[b_index] |= b_mask >> (slot & 7);
-        spin_unlock_irqrestore(&ps3_bolttab_lock, flags);
+        spin_unlock_irqrestore(&ps3_htab_lock, flags);
 
-        return (slot & 7) | (secondary << 3);
+        return ret;
 }
 
 static long ps3_hpte_remove(unsigned long hpte_group)
@@ -172,39 +109,48 @@ static long ps3_hpte_remove(unsigned long hpte_group)
 static long ps3_hpte_updatepp(unsigned long slot, unsigned long newpp,
         unsigned long va, int psize, int ssize, int local)
 {
+        int result;
+        u64 hpte_v, want_v, hpte_rs;
+        u64 hpte_v_array[4];
         unsigned long flags;
-        unsigned long result;
-        unsigned long pteg, bit;
-        unsigned long hpte_v, want_v;
+        long ret;
 
-        want_v = hpte_encode_v(va, psize, MMU_SEGSIZE_256M);
+        want_v = hpte_encode_v(va, psize, ssize);
 
-        spin_lock_irqsave(&ps3_bolttab_lock, flags);
+        spin_lock_irqsave(&ps3_htab_lock, flags);
 
-        hpte_v = htab[slot].v;
-        if (!HPTE_V_COMPARE(hpte_v, want_v) || !(hpte_v & HPTE_V_VALID)) {
-                spin_unlock_irqrestore(&ps3_bolttab_lock, flags);
-
-                /* ps3_hpte_insert() will be used to update PTE */
-                return -1;
-        }
-
-        result = lv1_write_htab_entry(0, slot, 0, 0);
+        result = lv1_read_htab_entries(PS3_LPAR_VAS_ID_CURRENT, slot & ~0x3UL,
+                                       &hpte_v_array[0], &hpte_v_array[1],
+                                       &hpte_v_array[2], &hpte_v_array[3],
+                                       &hpte_rs);
 
         if (result) {
-                DBG("%s: va=%lx slot=%lx psize=%d result = %ld (0x%lx)\n",
-                       __func__, va, slot, psize, result, result);
+                pr_info("%s: res=%d read va=%lx slot=%lx psize=%d\n",
+                        __func__, result, va, slot, psize);
                 BUG();
         }
 
-        pteg = slot / HPTES_PER_GROUP;
-        bit = slot % HPTES_PER_GROUP;
-        inusetab[pteg] &= ~(0x80 >> bit);
+        hpte_v = hpte_v_array[slot % 4];
 
-        spin_unlock_irqrestore(&ps3_bolttab_lock, flags);
+        /*
+         * As lv1_read_htab_entries() does not give us the RPN, we can
+         * not synthesize the new hpte_r value here, and therefore can
+         * not update the hpte with lv1_insert_htab_entry(), so we
+         * insted invalidate it and ask the caller to update it via
+         * ps3_hpte_insert() by returning a -1 value.
+         */
+        if (!HPTE_V_COMPARE(hpte_v, want_v) || !(hpte_v & HPTE_V_VALID)) {
+                /* not found */
+                ret = -1;
+        } else {
+                /* entry found, just invalidate it */
+                result = lv1_write_htab_entry(PS3_LPAR_VAS_ID_CURRENT,
+                                              slot, 0, 0);
+                ret = -1;
+        }
 
-        /* ps3_hpte_insert() will be used to update PTE */
-        return -1;
+        spin_unlock_irqrestore(&ps3_htab_lock, flags);
+        return ret;
 }
 
 static void ps3_hpte_updateboltedpp(unsigned long newpp, unsigned long ea,
@@ -217,45 +163,35 @@ static void ps3_hpte_invalidate(unsigned long slot, unsigned long va,
         int psize, int ssize, int local)
 {
         unsigned long flags;
-        unsigned long result;
-        unsigned long pteg, bit;
+        int result;
+
+        spin_lock_irqsave(&ps3_htab_lock, flags);
 
-        spin_lock_irqsave(&ps3_bolttab_lock, flags);
-        result = lv1_write_htab_entry(0, slot, 0, 0);
+        result = lv1_write_htab_entry(PS3_LPAR_VAS_ID_CURRENT, slot, 0, 0);
 
         if (result) {
-                DBG("%s: va=%lx slot=%lx psize=%d result = %ld (0x%lx)\n",
-                       __func__, va, slot, psize, result, result);
+                pr_info("%s: res=%d va=%lx slot=%lx psize=%d\n",
+                        __func__, result, va, slot, psize);
                 BUG();
         }
 
-        pteg = slot / HPTES_PER_GROUP;
-        bit = slot % HPTES_PER_GROUP;
-        inusetab[pteg] &= ~(0x80 >> bit);
-        spin_unlock_irqrestore(&ps3_bolttab_lock, flags);
+        spin_unlock_irqrestore(&ps3_htab_lock, flags);
 }
 
 static void ps3_hpte_clear(void)
 {
-        int result;
-
-        DBG(" -> %s:%d\n", __func__, __LINE__);
+        unsigned long hpte_count = (1UL << ppc64_pft_size) >> 4;
+        u64 i;
 
-        result = lv1_unmap_htab(htab_addr);
-        BUG_ON(result);
+        for (i = 0; i < hpte_count; i++)
+                lv1_write_htab_entry(PS3_LPAR_VAS_ID_CURRENT, i, 0, 0);
 
         ps3_mm_shutdown();
         ps3_mm_vas_destroy();
-
-        DBG(" <- %s:%d\n", __func__, __LINE__);
 }
 
 void __init ps3_hpte_init(unsigned long htab_size)
 {
-        long bitmap_size;
-
-        DBG(" -> %s:%d\n", __func__, __LINE__);
-
         ppc_md.hpte_invalidate = ps3_hpte_invalidate;
         ppc_md.hpte_updatepp = ps3_hpte_updatepp;
         ppc_md.hpte_updateboltedpp = ps3_hpte_updateboltedpp;
@@ -264,28 +200,5 @@ void __init ps3_hpte_init(unsigned long htab_size)
         ppc_md.hpte_clear_all = ps3_hpte_clear;
 
         ppc64_pft_size = __ilog2(htab_size);
-
-        bitmap_size = htab_size / sizeof(struct hash_pte) / 8;
-
-        bolttab = __va(lmb_alloc(bitmap_size, 1));
-        inusetab = __va(lmb_alloc(bitmap_size, 1));
-
-        memset(bolttab, 0, bitmap_size);
-        memset(inusetab, 0, bitmap_size);
-
-        DBG(" <- %s:%d\n", __func__, __LINE__);
 }
 
-void __init ps3_map_htab(void)
-{
-        long result;
-        unsigned long htab_size = (1UL << ppc64_pft_size);
-
-        result = lv1_map_htab(0, &htab_addr);
-
-        htab = (__force struct hash_pte *)ioremap_flags(htab_addr, htab_size,
-                                            pgprot_val(PAGE_READONLY_X));
-
-        DBG("%s:%d: lpar %016lxh, virt %016lxh\n", __func__, __LINE__,
-                htab_addr, (unsigned long)htab);
-}
diff --git a/arch/powerpc/platforms/ps3/setup.c b/arch/powerpc/platforms/ps3/setup.c
index a413abbd4123..77bc330263c4 100644
--- a/arch/powerpc/platforms/ps3/setup.c
+++ b/arch/powerpc/platforms/ps3/setup.c
@@ -201,7 +201,6 @@ static void __init ps3_setup_arch(void)
                ps3_firmware_version.rev);
 
         ps3_spu_set_platform();
-        ps3_map_htab();
 
 #ifdef CONFIG_SMP
         smp_init_ps3();
diff --git a/arch/powerpc/platforms/ps3/spu.c b/arch/powerpc/platforms/ps3/spu.c
index d135cef9ed6a..ccae3d446b98 100644
--- a/arch/powerpc/platforms/ps3/spu.c
+++ b/arch/powerpc/platforms/ps3/spu.c
@@ -186,14 +186,24 @@ static void spu_unmap(struct spu *spu)
         iounmap(spu_pdata(spu)->shadow);
 }
 
+/**
+ * setup_areas - Map the spu regions into the address space.
+ *
+ * The current HV requires the spu shadow regs to be mapped with the
+ * PTE page protection bits set as read-only (PP=3).  This implementation
+ * uses the low level __ioremap() to bypass the page protection settings
+ * inforced by ioremap_flags() to get the needed PTE bits set for the
+ * shadow regs.
+ */
+
 static int __init setup_areas(struct spu *spu)
 {
         struct table {char* name; unsigned long addr; unsigned long size;};
+        static const unsigned long shadow_flags = _PAGE_NO_CACHE | 3;
 
-        spu_pdata(spu)->shadow = ioremap_flags(spu_pdata(spu)->shadow_addr,
-                                               sizeof(struct spe_shadow),
-                                               pgprot_val(PAGE_READONLY) |
-                                               _PAGE_NO_CACHE);
+        spu_pdata(spu)->shadow = __ioremap(spu_pdata(spu)->shadow_addr,
+                                           sizeof(struct spe_shadow),
+                                           shadow_flags);
         if (!spu_pdata(spu)->shadow) {
                 pr_debug("%s:%d: ioremap shadow failed\n", __func__, __LINE__);
                 goto fail_ioremap;
diff --git a/arch/powerpc/platforms/pseries/plpar_wrappers.h b/arch/powerpc/platforms/pseries/plpar_wrappers.h
index a437267c6bf8..d967c1893ab5 100644
--- a/arch/powerpc/platforms/pseries/plpar_wrappers.h
+++ b/arch/powerpc/platforms/pseries/plpar_wrappers.h
@@ -2,6 +2,7 @@
 #define _PSERIES_PLPAR_WRAPPERS_H
 
 #include <asm/hvcall.h>
+#include <asm/page.h>
 
 static inline long poll_pending(void)
 {
@@ -44,12 +45,34 @@ static inline long register_slb_shadow(unsigned long cpu, unsigned long vpa)
 
 static inline long plpar_page_set_loaned(unsigned long vpa)
 {
-        return plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_LOANED, vpa, 0);
+        unsigned long cmo_page_sz = cmo_get_page_size();
+        long rc = 0;
+        int i;
+
+        for (i = 0; !rc && i < PAGE_SIZE; i += cmo_page_sz)
+                rc = plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_LOANED, vpa + i, 0);
+
+        for (i -= cmo_page_sz; rc && i != 0; i -= cmo_page_sz)
+                plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_ACTIVE,
+                                   vpa + i - cmo_page_sz, 0);
+
+        return rc;
 }
 
 static inline long plpar_page_set_active(unsigned long vpa)
 {
-        return plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_ACTIVE, vpa, 0);
+        unsigned long cmo_page_sz = cmo_get_page_size();
+        long rc = 0;
+        int i;
+
+        for (i = 0; !rc && i < PAGE_SIZE; i += cmo_page_sz)
+                rc = plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_ACTIVE, vpa + i, 0);
+
+        for (i -= cmo_page_sz; rc && i != 0; i -= cmo_page_sz)
+                plpar_hcall_norets(H_PAGE_INIT, H_PAGE_SET_LOANED,
+                                   vpa + i - cmo_page_sz, 0);
+
+        return rc;
 }
 
 extern void vpa_init(int cpu);
diff --git a/arch/powerpc/platforms/pseries/setup.c b/arch/powerpc/platforms/pseries/setup.c
index 063a0d2fba30..7b01d67b4e48 100644
--- a/arch/powerpc/platforms/pseries/setup.c
+++ b/arch/powerpc/platforms/pseries/setup.c
@@ -68,6 +68,10 @@
 #include "plpar_wrappers.h"
 #include "pseries.h"
 
+int CMO_PrPSP = -1;
+int CMO_SecPSP = -1;
+unsigned long CMO_PageSize = (ASM_CONST(1) << IOMMU_PAGE_SHIFT);
+EXPORT_SYMBOL(CMO_PageSize);
 
 int fwnmi_active;  /* TRUE if an FWNMI handler is present */
 
@@ -325,8 +329,7 @@ void pSeries_cmo_feature_init(void)
 {
         char *ptr, *key, *value, *end;
         int call_status;
-        int PrPSP = -1;
-        int SecPSP = -1;
+        int page_order = IOMMU_PAGE_SHIFT;
 
         pr_debug(" -> fw_cmo_feature_init()\n");
         spin_lock(&rtas_data_buf_lock);
@@ -365,21 +368,31 @@ void pSeries_cmo_feature_init(void)
                                 break;
                         }
 
-                        if (0 == strcmp(key, "PrPSP"))
-                                PrPSP = simple_strtol(value, NULL, 10);
+                        if (0 == strcmp(key, "CMOPageSize"))
+                                page_order = simple_strtol(value, NULL, 10);
+                        else if (0 == strcmp(key, "PrPSP"))
+                                CMO_PrPSP = simple_strtol(value, NULL, 10);
                         else if (0 == strcmp(key, "SecPSP"))
-                                SecPSP = simple_strtol(value, NULL, 10);
+                                CMO_SecPSP = simple_strtol(value, NULL, 10);
                         value = key = ptr + 1;
                 }
                 ptr++;
         }
 
-        if (PrPSP != -1 || SecPSP != -1) {
+        /* Page size is returned as the power of 2 of the page size,
+         * convert to the page size in bytes before returning
+         */
+        CMO_PageSize = 1 << page_order;
+        pr_debug("CMO_PageSize = %lu\n", CMO_PageSize);
+
+        if (CMO_PrPSP != -1 || CMO_SecPSP != -1) {
                 pr_info("CMO enabled\n");
-                pr_debug("CMO enabled, PrPSP=%d, SecPSP=%d\n", PrPSP, SecPSP);
+                pr_debug("CMO enabled, PrPSP=%d, SecPSP=%d\n", CMO_PrPSP,
+                         CMO_SecPSP);
                 powerpc_firmware_features |= FW_FEATURE_CMO;
         } else
-                pr_debug("CMO not enabled, PrPSP=%d, SecPSP=%d\n", PrPSP, SecPSP);
+                pr_debug("CMO not enabled, PrPSP=%d, SecPSP=%d\n", CMO_PrPSP,
+                         CMO_SecPSP);
         spin_unlock(&rtas_data_buf_lock);
         pr_debug(" <- fw_cmo_feature_init()\n");
 }