aboutsummaryrefslogtreecommitdiff
path: root/include/arch/pmem.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/arch/pmem.h')
-rw-r--r--include/arch/pmem.h6
1 files changed, 4 insertions, 2 deletions
diff --git a/include/arch/pmem.h b/include/arch/pmem.h
index 03a8172..6a17449 100644
--- a/include/arch/pmem.h
+++ b/include/arch/pmem.h
@@ -13,10 +13,12 @@
#include <kmi/mem.h> /* NUM_ORDERS */
#include <kmi/types.h>
-#if defined(riscv64)
+#if defined(__riscv)
+# if __riscv_xlen == 64
#include "../../arch/riscv64/include/pmem.h"
-#elif defined(riscv32)
+# else
#include "../../arch/riscv32/include/pmem.h"
+# endif
#endif
/**