1 files changed, 11 insertions, 3 deletions
diff --git a/arch/riscv64/kernel/pmem.c b/arch/riscv64/kernel/pmem.c
index e438131..3076dc5 100644
--- a/arch/riscv64/kernel/pmem.c
+++ b/arch/riscv64/kernel/pmem.c
@@ -12,13 +12,21 @@ stat_t stat_pmem_conf(void *fdt, size_t *max_order, size_t *base_bits,
                       size_t bits[NUM_ORDERS])
 {
 	UNUSED(fdt);
+	*base_bits = 12;
+
+	size_t nbits = 9;
 
-	/* Sv39 for now */
+#if defined(riscv64)
 	*max_order = 2;
-	*base_bits = 12;
+	nbits = 9;
+#else
+	*max_order = 1;
+	nbits = 10;
+#endif
+
 	/* we can assume bits[] is zeroed out beforehand */
 	for (size_t i = 0; i <= *max_order; ++i)
-		bits[i] = 9;
+		bits[i] = nbits;
 
 	return OK;
 }