Skip to content

Commit 9639de5

Browse files
committed
mlkem: ref: poly ntt/invntt scheduling
1 parent 7046ee9 commit 9639de5

File tree

1 file changed

+4
-4
lines changed
  • src/crypto_kem/mlkem/mlkem768/amd64/ref

1 file changed

+4
-4
lines changed

src/crypto_kem/mlkem/mlkem768/amd64/ref/poly.jinc

+4-4
Original file line numberDiff line numberDiff line change
@@ -486,9 +486,9 @@ fn _poly_invntt(reg ptr u16[MLKEM_N] rp) -> reg ptr u16[MLKEM_N]
486486
cmp = start; cmp += len;
487487
while (j < cmp)
488488
{
489+
t = rp[(int)j];
489490
offset = j; offset += len;
490491
s = rp[(int)offset];
491-
t = rp[(int)j];
492492
m = s; m += t;
493493
m = __barrett_reduce(m);
494494
rp[(int)j] = m;
@@ -544,14 +544,14 @@ fn _poly_ntt(reg ptr u16[MLKEM_N] rp) -> reg ptr u16[MLKEM_N]
544544
cmp = start; cmp += len;
545545
while (j < cmp)
546546
{
547+
s = rp[(int)j];
548+
m = s;
547549
offset = j; offset += len;
548550
t = rp[(int)offset];
549551
t = __fqmul(t, zeta);
550-
s = rp[(int)j];
551-
m = s;
552552
m -= t;
553-
rp[(int)offset] = m;
554553
t += s;
554+
rp[(int)offset] = m;
555555
rp[(int)j] = t;
556556
j += 1;
557557
}

0 commit comments

Comments
 (0)