#define cpu_switch_mm(pgd,mm) cpu_do_switch_mm(virt_to_phys(pgd),mm)
 
 #ifdef CONFIG_ARM_LPAE
+
+#define cpu_get_ttbr(nr)                                       \
+       ({                                                      \
+               u64 ttbr;                                       \
+               __asm__("mrrc   p15, " #nr ", %Q0, %R0, c2"     \
+                       : "=r" (ttbr));                         \
+               ttbr;                                           \
+       })
+
+#define cpu_set_ttbr(nr, val)                                  \
+       do {                                                    \
+               u64 ttbr = val;                                 \
+               __asm__("mcrr   p15, " #nr ", %Q0, %R0, c2"     \
+                       : : "r" (ttbr));                        \
+       } while (0)
+
 #define cpu_get_pgd()  \
        ({                                              \
-               unsigned long pg, pg2;                  \
-               __asm__("mrrc   p15, 0, %0, %1, c2"     \
-                       : "=r" (pg), "=r" (pg2)         \
-                       :                               \
-                       : "cc");                        \
+               u64 pg = cpu_get_ttbr(0);               \
                pg &= ~(PTRS_PER_PGD*sizeof(pgd_t)-1);  \
                (pgd_t *)phys_to_virt(pg);              \
        })
 
 #include <asm/smp_plat.h>
 #include <asm/thread_notify.h>
 #include <asm/tlbflush.h>
+#include <asm/proc-fns.h>
 
 /*
  * On ARMv6, we have the following structure in the Context ID:
 #ifdef CONFIG_ARM_LPAE
 static void cpu_set_reserved_ttbr0(void)
 {
-       unsigned long ttbl = __pa(swapper_pg_dir);
-       unsigned long ttbh = 0;
-
        /*
         * Set TTBR0 to swapper_pg_dir which contains only global entries. The
         * ASID is set to 0.
         */
-       asm volatile(
-       "       mcrr    p15, 0, %0, %1, c2              @ set TTBR0\n"
-       :
-       : "r" (ttbl), "r" (ttbh));
+       cpu_set_ttbr(0, __pa(swapper_pg_dir));
        isb();
 }
 #else