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