wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2026-07-05 17:00:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #10430 from SparkiDev/mlkem_avx2_fixes

Browse Source 
ML-KEM: fix AVX2 assembly
This commit is contained in:
David Garske
2026-05-12 09:25:54 -07:00
committed by GitHub
parent f8bc0ce95a 3c9423257f
commit 2239fc336b
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wolfcrypt/src/wc_mlkem.c
+2
View File
@@ -185,6 +185,7 @@ sword16 wc_mlkem_opt_blocker(void) {
/******************************************************************************/
#ifndef WOLFSSL_MLKEM_NO_MAKE_KEY
/* Get the k value (number of polynomials in a vector) from the key type.
*
* @param [in] key ML-KEM key object.
@@ -225,6 +226,7 @@ static int mlkemkey_get_k(const MlKemKey* key)
return 0;
}
}
#endif
#ifdef WOLFSSL_MLKEM_DYNAMIC_KEYS
/* Allocate (or reallocate) the private key buffer, right-sized for k.
wolfcrypt/src/wc_mlkem_asm.S
+5 -2
View File
@@ -14458,8 +14458,8 @@ L_mlkem_decompress_5_avx2_shuf:
.p2align 4
#endif /* __APPLE__ */
L_mlkem_decompress_5_avx2_mask:
.short 0x001f,0x03e0,0x007c,0x0f80,0x01f0,0x003e,0x07c0,0x00fb
.short 0x001f,0x03e0,0x007c,0x0f80,0x01f0,0x003e,0x07c0,0x00fb
.short 0x001f,0x03e0,0x007c,0x0f80,0x01f0,0x003e,0x07c0,0x00f8
.short 0x001f,0x03e0,0x007c,0x0f80,0x01f0,0x003e,0x07c0,0x00f8
#ifndef __APPLE__
.data
#else
@@ -15534,6 +15534,9 @@ _mlkem_cmp_avx2:
vpxor 1504(%rsi), %ymm1, %ymm1
vpor %ymm0, %ymm2, %ymm2
vpor %ymm1, %ymm3, %ymm3
vmovdqu 1536(%rdi), %ymm0
vpxor 1536(%rsi), %ymm0, %ymm0
vpor %ymm0, %ymm2, %ymm2
L_mlkem_cmp_avx2_done:
vpor %ymm3, %ymm2, %ymm2
vptest %ymm2, %ymm2