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