Skip to content

Commit dd63671

Browse files
committed
1 parent 76e3b53 commit dd63671

File tree

1 file changed

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

1 file changed

+7
-13
lines changed

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

+7-13
Original file line numberDiff line numberDiff line change
@@ -176,23 +176,21 @@ fn _i_poly_compress(reg ptr u8[MLKEM_POLYCOMPRESSEDBYTES] rp, reg ptr u16[MLKEM_
176176
{
177177
reg u16 t;
178178
reg u32 d0, d1;
179-
reg u64 i j;
179+
reg u64 i;
180180

181181
a = _poly_csubq(a);
182182

183183
i = 0;
184-
j = 0;
185184
while(i < 128)
186185
{
187-
t = a[(int)j];
186+
t = a[2 * i];
188187
d0 = (32u)t;
189188
d0 <<= 4;
190189
d0 += 1665;
191190
d0 *= 80635;
192191
d0 >>= 28;
193192
d0 &= 0xf;
194-
j += 1;
195-
t = a[(int)j];
193+
t = a[2 * i + 1];
196194
d1 = (32u)t;
197195
d1 <<= 4;
198196
d1 += 1665;
@@ -201,9 +199,8 @@ fn _i_poly_compress(reg ptr u8[MLKEM_POLYCOMPRESSEDBYTES] rp, reg ptr u16[MLKEM_
201199
d1 &= 0xf;
202200
d1 <<= 4;
203201
d0 |= d1;
204-
rp[(int) i] = d0;
202+
rp[i] = d0;
205203
i += 1;
206-
j += 1;
207204
}
208205
return rp, a;
209206
}
@@ -213,10 +210,9 @@ fn _poly_decompress(reg ptr u16[MLKEM_N] rp, reg u64 ap) -> stack u16[MLKEM_N]
213210
{
214211
reg u8 t;
215212
reg u16 d0, d1;
216-
reg u64 i j;
213+
reg u64 i;
217214

218215
i = 0;
219-
j = 0;
220216

221217
while (i < 128) {
222218
t = (u8)[ap+i];
@@ -230,10 +226,8 @@ fn _poly_decompress(reg ptr u16[MLKEM_N] rp, reg u64 ap) -> stack u16[MLKEM_N]
230226
d1 += 8;
231227
d0 >>= 4;
232228
d1 >>= 4;
233-
rp[(int)j] = d0;
234-
j += 1;
235-
rp[(int)j] = d1;
236-
j += 1;
229+
rp[2 * i] = d0;
230+
rp[2 * i + 1] = d1;
237231
i += 1;
238232
}
239233
return rp;

0 commit comments

Comments
 (0)