mlkem786: merge add

formosa-crypto · Feb 7, 2024 · 37c9502 · 37c9502
1 parent 318662c
commit 37c9502
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/src/crypto_kem/mlkem/mlkem768/amd64/avx2/kem.jinc b/src/crypto_kem/mlkem/mlkem768/amd64/avx2/kem.jinc
@@ -106,9 +106,8 @@ fn __crypto_kem_dec_jazz(reg u64 shkp, reg u64 ctp, reg u64 skp)
 
   buf[0:MLKEM_INDCPA_MSGBYTES] = __indcpa_dec_1(buf[0:MLKEM_INDCPA_MSGBYTES], ctp, skp);
 
-  //hp = #LEA(skp + 32);
-  hp = skp; hp += 32;
-  hp += 24 * MLKEM_K * MLKEM_N>>3;
+  hp = skp;
+  hp += 32 + (24 * MLKEM_K * MLKEM_N>>3);
 
   /* fixme: should loads be 256-bits long? */
   for i=0 to MLKEM_SYMBYTES/8

diff --git a/src/crypto_kem/mlkem/mlkem768/amd64/ref/kem.jinc b/src/crypto_kem/mlkem/mlkem768/amd64/ref/kem.jinc
@@ -107,9 +107,8 @@ fn __crypto_kem_dec_jazz(reg u64 shkp, reg u64 ctp, reg u64 skp)
 
   buf[0:MLKEM_MSGBYTES] = __indcpa_dec(buf[0:MLKEM_MSGBYTES], ctp, skp);
 
-  //hp = #LEA(skp + 32);
-  hp = skp; hp += 32;
-  hp += 24 * MLKEM_K * MLKEM_N>>3;
+  hp = skp;
+  hp += 32 + (24 * MLKEM_K * MLKEM_N>>3);
 
   for i=0 to MLKEM_SYMBYTES/8
   {