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