#define HGATP_MODE_SV32X4      _AC(1, UL)
 #define HGATP_MODE_SV39X4      _AC(8, UL)
 #define HGATP_MODE_SV48X4      _AC(9, UL)
+#define HGATP_MODE_SV57X4      _AC(10, UL)
 
 #define HGATP32_MODE_SHIFT     31
 #define HGATP32_VMID_SHIFT     22
 
 void kvm_riscv_gstage_mode_detect(void)
 {
 #ifdef CONFIG_64BIT
+       /* Try Sv57x4 G-stage mode */
+       csr_write(CSR_HGATP, HGATP_MODE_SV57X4 << HGATP_MODE_SHIFT);
+       if ((csr_read(CSR_HGATP) >> HGATP_MODE_SHIFT) == HGATP_MODE_SV57X4) {
+               gstage_mode = (HGATP_MODE_SV57X4 << HGATP_MODE_SHIFT);
+               gstage_pgd_levels = 5;
+               goto skip_sv48x4_test;
+       }
+
        /* Try Sv48x4 G-stage mode */
        csr_write(CSR_HGATP, HGATP_MODE_SV48X4 << HGATP_MODE_SHIFT);
        if ((csr_read(CSR_HGATP) >> HGATP_MODE_SHIFT) == HGATP_MODE_SV48X4) {
                gstage_mode = (HGATP_MODE_SV48X4 << HGATP_MODE_SHIFT);
                gstage_pgd_levels = 4;
        }
-       csr_write(CSR_HGATP, 0);
+skip_sv48x4_test:
 
+       csr_write(CSR_HGATP, 0);
        __kvm_riscv_hfence_gvma_all();
 #endif
 }