logos-storage/nim-goldilocks-hash
1
0
Fork 0
mirror of https://github.com/logos-storage/nim-goldilocks-hash.git synced 2026-01-09 09:03:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

a small optimization in the poseidon2 hash

Browse Source
This commit is contained in:
Balazs Komuves 2024-10-03 00:32:49 +02:00
parent d89a5cca7b
commit 06bd13b2bd
No known key found for this signature in database
GPG Key ID: F63B7AEF18435562
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cbits/goldilocks.c
View File

@ -17,6 +17,11 @@ uint64_t goldilocks_add(uint64_t x, uint64_t y) {
return ( (z >= GOLDILOCKS_PRIME) || (z<x) ) ? (z - GOLDILOCKS_PRIME) : z;
}
uint64_t goldilocks_add_to_uint64(uint64_t x, uint64_t y) {
uint64_t z = x + y;
return (z<x) ? (z - GOLDILOCKS_PRIME) : z;
}
uint64_t goldilocks_sub(uint64_t x, uint64_t y) {
uint64_t z = x - y;
return (z > x) ? (z + GOLDILOCKS_PRIME) : z;
@ -80,6 +85,11 @@ uint64_t goldilocks_mul(uint64_t x, uint64_t y) {
return goldilocks_rdc(z);
}
uint64_t goldilocks_mul_to_uint64(uint64_t x, uint64_t y) {
__uint128_t z = (__uint128_t)x * (__uint128_t)y;
return goldilocks_rdc_to_uint64(z);
}
uint64_t goldilocks_mul_add128(uint64_t x, uint64_t y, __uint128_t z) {
__uint128_t w = (__uint128_t)x * (__uint128_t)y + z;
return goldilocks_rdc(w);
@ -147,11 +157,11 @@ from <https://github.com/HorizenLabs/poseidon2/blob/main/plain_implementations/s
*/
uint64_t goldilocks_poseidon2_sbox(uint64_t x0, uint64_t rc) {
uint64_t x = goldilocks_add( x0 , rc );
uint64_t x2 = goldilocks_sqr( x );
uint64_t x4 = goldilocks_sqr( x2 );
uint64_t x6 = goldilocks_mul( x4 , x2 );
uint64_t x7 = goldilocks_mul( x6 , x );
uint64_t x = goldilocks_add_to_uint64( x0 , rc );
uint64_t x2 = goldilocks_mul_to_uint64( x , x );
uint64_t x4 = goldilocks_mul_to_uint64( x2 , x2 );
uint64_t x6 = goldilocks_mul_to_uint64( x4 , x2 );
uint64_t x7 = goldilocks_mul ( x6 , x );
return x7;
}

1
goldilocks_hash/goldilocks.nim
View File

@ -10,4 +10,3 @@ func `*`* (x, y: F): F {. header: "../cbits/goldilocks.h", importc: "goldilocks_
proc `+=`* (x: var F, y: F) = x = x + y
proc `-=`* (x: var F, y: F) = x = x - y
proc `*=`* (x: var F, y: F) = x = x * y