#define KVM_REG_MIPS_CP0_CONFIG3       MIPS_CP0_32(16, 3)
 #define KVM_REG_MIPS_CP0_CONFIG4       MIPS_CP0_32(16, 4)
 #define KVM_REG_MIPS_CP0_CONFIG5       MIPS_CP0_32(16, 5)
+#define KVM_REG_MIPS_CP0_CONFIG6       MIPS_CP0_32(16, 6)
 #define KVM_REG_MIPS_CP0_CONFIG7       MIPS_CP0_32(16, 7)
 #define KVM_REG_MIPS_CP0_MAARI         MIPS_CP0_64(17, 2)
 #define KVM_REG_MIPS_CP0_XCONTEXT      MIPS_CP0_64(20, 0)
+#define KVM_REG_MIPS_CP0_DIAG          MIPS_CP0_32(22, 0)
 #define KVM_REG_MIPS_CP0_ERROREPC      MIPS_CP0_64(30, 0)
 #define KVM_REG_MIPS_CP0_KSCRATCH1     MIPS_CP0_64(31, 2)
 #define KVM_REG_MIPS_CP0_KSCRATCH2     MIPS_CP0_64(31, 3)
 #define MIPS_CP0_WATCH_LO      18
 #define MIPS_CP0_WATCH_HI      19
 #define MIPS_CP0_TLB_XCONTEXT  20
+#define MIPS_CP0_DIAG          22
 #define MIPS_CP0_ECC           26
 #define MIPS_CP0_CACHE_ERR     27
 #define MIPS_CP0_TAG_LO                28
                          unsigned int count);
 void kvm_vz_load_guesttlb(const struct kvm_mips_tlb *buf, unsigned int index,
                          unsigned int count);
+#ifdef CONFIG_CPU_LOONGSON64
+void kvm_loongson_clear_guest_vtlb(void);
+void kvm_loongson_clear_guest_ftlb(void);
+#endif
 #endif
 
 void kvm_mips_suspend_mm(int cpu);
 
 
 #include <asm/cpu.h>
 #include <asm/bootinfo.h>
+#include <asm/mipsregs.h>
 #include <asm/mmu_context.h>
 #include <asm/pgtable.h>
 #include <asm/cacheflush.h>
 }
 EXPORT_SYMBOL_GPL(kvm_vz_load_guesttlb);
 
+#ifdef CONFIG_CPU_LOONGSON64
+void kvm_loongson_clear_guest_vtlb(void)
+{
+       int idx = read_gc0_index();
+
+       /* Set root GuestID for root probe and write of guest TLB entry */
+       set_root_gid_to_guest_gid();
+
+       write_gc0_index(0);
+       guest_tlbinvf();
+       write_gc0_index(idx);
+
+       clear_root_gid();
+       set_c0_diag(LOONGSON_DIAG_ITLB | LOONGSON_DIAG_DTLB);
+}
+EXPORT_SYMBOL_GPL(kvm_loongson_clear_guest_vtlb);
+
+void kvm_loongson_clear_guest_ftlb(void)
+{
+       int i;
+       int idx = read_gc0_index();
+
+       /* Set root GuestID for root probe and write of guest TLB entry */
+       set_root_gid_to_guest_gid();
+
+       for (i = current_cpu_data.tlbsizevtlb;
+            i < (current_cpu_data.tlbsizevtlb +
+                    current_cpu_data.tlbsizeftlbsets);
+            i++) {
+               write_gc0_index(i);
+               guest_tlbinvf();
+       }
+       write_gc0_index(idx);
+
+       clear_root_gid();
+       set_c0_diag(LOONGSON_DIAG_ITLB | LOONGSON_DIAG_DTLB);
+}
+EXPORT_SYMBOL_GPL(kvm_loongson_clear_guest_ftlb);
+#endif
+
 #endif
 
 /**
 
        return mask;
 }
 
+static inline unsigned int kvm_vz_config6_guest_wrmask(struct kvm_vcpu *vcpu)
+{
+       return MIPS_CONF6_LOONGSON_INTIMER | MIPS_CONF6_LOONGSON_EXTIMER;
+}
+
 /*
  * VZ optionally allows these additional Config bits to be written by root:
  * Config:     M, [MT]
        return kvm_vz_config5_guest_wrmask(vcpu) | MIPS_CONF5_MRP;
 }
 
+static inline unsigned int kvm_vz_config6_user_wrmask(struct kvm_vcpu *vcpu)
+{
+       return kvm_vz_config6_guest_wrmask(vcpu) |
+               MIPS_CONF6_LOONGSON_SFBEN | MIPS_CONF6_LOONGSON_FTLBDIS;
+}
+
 static gpa_t kvm_vz_gva_to_gpa_cb(gva_t gva)
 {
        /* VZ guest has already converted gva to gpa */
                                    (sel == 2 ||        /* SRSCtl */
                                     sel == 3)) ||      /* SRSMap */
                                   (rd == MIPS_CP0_CONFIG &&
-                                   (sel == 7)) ||      /* Config7 */
+                                   (sel == 6 ||        /* Config6 */
+                                    sel == 7)) ||      /* Config7 */
                                   (rd == MIPS_CP0_LLADDR &&
                                    (sel == 2) &&       /* MAARI */
                                    cpu_guest_has_maar &&
                                   (rd == MIPS_CP0_ERRCTL &&
                                    (sel == 0))) {      /* ErrCtl */
                                val = cop0->reg[rd][sel];
+#ifdef CONFIG_CPU_LOONGSON64
+                       } else if (rd == MIPS_CP0_DIAG &&
+                                  (sel == 0)) {        /* Diag */
+                               val = cop0->reg[rd][sel];
+#endif
                        } else {
                                val = 0;
                                er = EMULATE_FAIL;
                                   cpu_guest_has_maar &&
                                   !cpu_guest_has_dyn_maar) {
                                kvm_write_maari(vcpu, val);
+                       } else if (rd == MIPS_CP0_CONFIG &&
+                                  (sel == 6)) {
+                               cop0->reg[rd][sel] = (int)val;
                        } else if (rd == MIPS_CP0_ERRCTL &&
                                   (sel == 0)) {        /* ErrCtl */
                                /* ignore the written value */
+#ifdef CONFIG_CPU_LOONGSON64
+                       } else if (rd == MIPS_CP0_DIAG &&
+                                  (sel == 0)) {        /* Diag */
+                               unsigned long flags;
+
+                               local_irq_save(flags);
+                               if (val & LOONGSON_DIAG_BTB) {
+                                       /* Flush BTB */
+                                       set_c0_diag(LOONGSON_DIAG_BTB);
+                               }
+                               if (val & LOONGSON_DIAG_ITLB) {
+                                       /* Flush ITLB */
+                                       set_c0_diag(LOONGSON_DIAG_ITLB);
+                               }
+                               if (val & LOONGSON_DIAG_DTLB) {
+                                       /* Flush DTLB */
+                                       set_c0_diag(LOONGSON_DIAG_DTLB);
+                               }
+                               if (val & LOONGSON_DIAG_VTLB) {
+                                       /* Flush VTLB */
+                                       kvm_loongson_clear_guest_vtlb();
+                               }
+                               if (val & LOONGSON_DIAG_FTLB) {
+                                       /* Flush FTLB */
+                                       kvm_loongson_clear_guest_ftlb();
+                               }
+                               local_irq_restore(flags);
+#endif
                        } else {
                                er = EMULATE_FAIL;
                        }
        KVM_REG_MIPS_CP0_CONFIG3,
        KVM_REG_MIPS_CP0_CONFIG4,
        KVM_REG_MIPS_CP0_CONFIG5,
+       KVM_REG_MIPS_CP0_CONFIG6,
 #ifdef CONFIG_64BIT
        KVM_REG_MIPS_CP0_XCONTEXT,
 #endif
                        return -EINVAL;
                *v = read_gc0_config5();
                break;
+       case KVM_REG_MIPS_CP0_CONFIG6:
+               *v = kvm_read_sw_gc0_config6(cop0);
+               break;
        case KVM_REG_MIPS_CP0_MAAR(0) ... KVM_REG_MIPS_CP0_MAAR(0x3f):
                if (!cpu_guest_has_maar || cpu_guest_has_dyn_maar)
                        return -EINVAL;
                        write_gc0_config5(v);
                }
                break;
+       case KVM_REG_MIPS_CP0_CONFIG6:
+               cur = kvm_read_sw_gc0_config6(cop0);
+               change = (cur ^ v) & kvm_vz_config6_user_wrmask(vcpu);
+               if (change) {
+                       v = cur ^ change;
+                       kvm_write_sw_gc0_config6(cop0, (int)v);
+               }
+               break;
        case KVM_REG_MIPS_CP0_MAAR(0) ... KVM_REG_MIPS_CP0_MAAR(0x3f):
                if (!cpu_guest_has_maar || cpu_guest_has_dyn_maar)
                        return -EINVAL;