diff options
Diffstat (limited to 'cpukit')
-rw-r--r-- | cpukit/score/cpu/riscv/include/rtems/asm.h | 8 | ||||
-rw-r--r-- | cpukit/score/cpu/riscv/include/rtems/score/cpuimpl.h | 15 |
2 files changed, 23 insertions, 0 deletions
diff --git a/cpukit/score/cpu/riscv/include/rtems/asm.h b/cpukit/score/cpu/riscv/include/rtems/asm.h index 34b6474515..55fcb801c3 100644 --- a/cpukit/score/cpu/riscv/include/rtems/asm.h +++ b/cpukit/score/cpu/riscv/include/rtems/asm.h @@ -117,4 +117,12 @@ #define EXTERN(sym) .extern SYM (sym) #define TYPE_FUNC(sym) .type SYM (sym), %function +.macro GET_SELF_CPU_CONTROL REG +#ifdef RTEMS_SMP + csrr \REG, mscratch +#else + la \REG, _Per_CPU_Information +#endif +.endm + #endif diff --git a/cpukit/score/cpu/riscv/include/rtems/score/cpuimpl.h b/cpukit/score/cpu/riscv/include/rtems/score/cpuimpl.h index 2a63e03513..1370e656dd 100644 --- a/cpukit/score/cpu/riscv/include/rtems/score/cpuimpl.h +++ b/cpukit/score/cpu/riscv/include/rtems/score/cpuimpl.h @@ -52,6 +52,21 @@ extern "C" { #endif +#ifdef RTEMS_SMP + +static inline struct Per_CPU_Control *_RISCV_Get_current_per_CPU_control( void ) +{ + struct Per_CPU_Control *cpu_self; + + __asm__ volatile ( "csrr %0, mscratch" : "=r" ( cpu_self ) ); + + return cpu_self; +} + +#define _CPU_Get_current_per_CPU_control() _RISCV_Get_current_per_CPU_control() + +#endif /* RTEMS_SMP */ + #ifdef __cplusplus } #endif |