Optimize log256's binary search (#5284)

This commit is contained in:
cairo
2024-11-26 19:15:53 +01:00
committed by GitHub
parent 4afd599329
commit fdf7012d3b

View File

@ -644,29 +644,17 @@ library Math {
* *
* Adding one to the result gives the number of pairs of hex symbols needed to represent `value` as a hex string. * Adding one to the result gives the number of pairs of hex symbols needed to represent `value` as a hex string.
*/ */
function log256(uint256 value) internal pure returns (uint256) { function log256(uint256 x) internal pure returns (uint256 r) {
uint256 result = 0; // If value has upper 128 bits set, log2 result is at least 128
uint256 isGt; r = SafeCast.toUint(x > 0xffffffffffffffffffffffffffffffff) << 7;
unchecked { // If upper 64 bits of 128-bit half set, add 64 to result
isGt = SafeCast.toUint(value > (1 << 128) - 1); r |= SafeCast.toUint((x >> r) > 0xffffffffffffffff) << 6;
value >>= isGt * 128; // If upper 32 bits of 64-bit half set, add 32 to result
result += isGt * 16; r |= SafeCast.toUint((x >> r) > 0xffffffff) << 5;
// If upper 16 bits of 32-bit half set, add 16 to result
isGt = SafeCast.toUint(value > (1 << 64) - 1); r |= SafeCast.toUint((x >> r) > 0xffff) << 4;
value >>= isGt * 64; // Add 1 if upper 8 bits of 16-bit half set, and divide accumulated result by 8
result += isGt * 8; return (r >> 3) | SafeCast.toUint((x >> r) > 0xff);
isGt = SafeCast.toUint(value > (1 << 32) - 1);
value >>= isGt * 32;
result += isGt * 4;
isGt = SafeCast.toUint(value > (1 << 16) - 1);
value >>= isGt * 16;
result += isGt * 2;
result += SafeCast.toUint(value > (1 << 8) - 1);
}
return result;
} }
/** /**