Gas optimization on average function of Math.sol (#2757)

* change implementation to save gas

* add average test with two max uni256 number
This commit is contained in:
rotcivegaf
2021-07-10 21:28:12 +02:00
committed by GitHub
parent 1c1ebd765d
commit 6d97f09195
2 changed files with 7 additions and 2 deletions

View File

@ -25,8 +25,8 @@ library Math {
* zero.
*/
function average(uint256 a, uint256 b) internal pure returns (uint256) {
// (a + b) / 2 can overflow, so we distribute.
return (a / 2) + (b / 2) + (((a % 2) + (b % 2)) / 2);
// (a + b) / 2 can overflow.
return (a & b) + (a ^ b) / 2;
}
/**

View File

@ -54,6 +54,11 @@ contract('Math', function (accounts) {
const b = new BN('84346');
expect(await this.math.average(a, b)).to.be.bignumber.equal(bnAverage(a, b));
});
it('is correctly calculated with two max uint256 numbers', async function () {
const a = MAX_UINT256;
expect(await this.math.average(a, a)).to.be.bignumber.equal(bnAverage(a, a));
});
});
describe('ceilDiv', function () {