aboutsummaryrefslogtreecommitdiff
path: root/include/arch/tcb.h
diff options
context:
space:
mode:
Diffstat (limited to 'include/arch/tcb.h')
-rw-r--r--include/arch/tcb.h6
1 files changed, 4 insertions, 2 deletions
diff --git a/include/arch/tcb.h b/include/arch/tcb.h
index 22d3b59..931b800 100644
--- a/include/arch/tcb.h
+++ b/include/arch/tcb.h
@@ -9,10 +9,12 @@
* Arch-specific tcb stuff.
*/
-#if defined(riscv64)
+#if defined(__riscv)
+# if __riscv_xlen == 64
#include "../../arch/riscv64/include/tcb.h"
-#elif defined(riscv32)
+# else
#include "../../arch/riscv32/include/tcb.h"
+# endif
#endif
/**