diff --git a/wolfssl/wolfcrypt/wc_port.h b/wolfssl/wolfcrypt/wc_port.h index 1c0841924..b63c1a603 100644 --- a/wolfssl/wolfcrypt/wc_port.h +++ b/wolfssl/wolfcrypt/wc_port.h @@ -94,17 +94,17 @@ #include #include #if defined(WOLFSSL_AESNI) || defined(USE_INTEL_SPEEDUP) - #if LINUX_VERSION_CODE < KERNEL_VERSION(4, 0, 0) - #include - #else - #include - #endif - #define SAVE_VECTOR_REGISTERS() kernel_fpu_begin() - #define RESTORE_VECTOR_REGISTERS() kernel_fpu_end() + #if LINUX_VERSION_CODE < KERNEL_VERSION(4, 0, 0) + #include + #else + #include + #endif + #define SAVE_VECTOR_REGISTERS() kernel_fpu_begin() + #define RESTORE_VECTOR_REGISTERS() kernel_fpu_end() #elif defined(WOLFSSL_ARMASM) #include #define SAVE_VECTOR_REGISTERS() ({ preempt_disable(); fpsimd_preserve_current_state(); }) - #define SAVE_VECTOR_REGISTERS() ({ fpsimd_restore_current_state(); preempt_enable(); }) + #define RESTORE_VECTOR_REGISTERS() ({ fpsimd_restore_current_state(); preempt_enable(); }) #else #define SAVE_VECTOR_REGISTERS() ({}) #define RESTORE_VECTOR_REGISTERS() ({})