# update mstatus
csrr t1, CSR_MSTATUS
-#ifdef __riscv32
+#if XLEN == 32
li t0, (MSTATUS_MPRV | (VM_SV32 << 24))
#else
li t0, (MSTATUS_MPRV | (VM_SV39 << 24))
.balign 0x1000
page_table:
-#ifdef __riscv32
+#if XLEN == 32
.word ((0x80000000 >> 2) | PTE_V | PTE_R | PTE_W | PTE_X | PTE_G | PTE_U)
#else
.word ((0x80000000 >> 2) | PTE_V | PTE_R | PTE_W | PTE_X | PTE_G | PTE_U)