[SPARC64]: Patch up mmu context register writes for sun4v.

sun4v uses ASI_MMU instead of ASI_DMMU

Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 17 insertions, 13 deletions

diff --git a/arch/sparc64/kernel/setup.c b/arch/sparc64/kernel/setup.c
index 6d6178efd587..2d64320d3a4d 100644
--- a/arch/sparc64/kernel/setup.c
+++ b/arch/sparc64/kernel/setup.c
@@ -189,26 +189,30 @@ int prom_callback(long *args)
                 }
 
                 if ((va >= KERNBASE) && (va < (KERNBASE + (4 * 1024 * 1024)))) {
-                        extern unsigned long sparc64_kern_pri_context;
-
-                        /* Spitfire Errata #32 workaround */
-                        __asm__ __volatile__("stxa      %0, [%1] %2\n\t"
-                                             "flush     %%g6"
-                                             : /* No outputs */
-                                             : "r" (sparc64_kern_pri_context),
-                                               "r" (PRIMARY_CONTEXT),
-                                               "i" (ASI_DMMU));
+                        if (tlb_type == spitfire) {
+                                extern unsigned long sparc64_kern_pri_context;
+
+                                /* Spitfire Errata #32 workaround */
+                                __asm__ __volatile__(
+                                        "stxa   %0, [%1] %2\n\t"
+                                        "flush  %%g6"
+                                        : /* No outputs */
+                                        : "r" (sparc64_kern_pri_context),
+                                          "r" (PRIMARY_CONTEXT),
+                                          "i" (ASI_DMMU));
+                        }
 
                         /*
                          * Locked down tlb entry.
                          */
 
-                        if (tlb_type == spitfire)
+                        if (tlb_type == spitfire) {
                                 tte = spitfire_get_dtlb_data(SPITFIRE_HIGHEST_LOCKED_TLBENT);
-                        else if (tlb_type == cheetah || tlb_type == cheetah_plus)
+                                res = PROM_TRUE;
+                        } else if (tlb_type == cheetah || tlb_type == cheetah_plus) {
                                 tte = cheetah_get_ldtlb_data(CHEETAH_HIGHEST_LOCKED_TLBENT);
-
-                        res = PROM_TRUE;
+                                res = PROM_TRUE;
+                        }
                         goto done;
                 }