Commit 8c86fb68 authored by Uwe Kleine-König's avatar Uwe Kleine-König Committed by Andrew Morton

mul_u64_u64_div_u64: increase precision by conditionally swapping a and b

As indicated in the added comment, the algorithm works better if b is big.
As multiplication is commutative, a and b can be swapped.  Do this if a
is bigger than b.

Link: https://lkml.kernel.org/r/20240303092408.662449-2-u.kleine-koenig@pengutronix.deSigned-off-by: default avatarUwe Kleine-König <u.kleine-koenig@pengutronix.de>
Tested-by: default avatarBiju Das <biju.das.jz@bp.renesas.com>
Signed-off-by: default avatarAndrew Morton <akpm@linux-foundation.org>
parent 4bb7be96
...@@ -22,6 +22,7 @@ ...@@ -22,6 +22,7 @@
#include <linux/export.h> #include <linux/export.h>
#include <linux/math.h> #include <linux/math.h>
#include <linux/math64.h> #include <linux/math64.h>
#include <linux/minmax.h>
#include <linux/log2.h> #include <linux/log2.h>
/* Not needed on 64bit architectures */ /* Not needed on 64bit architectures */
...@@ -190,6 +191,20 @@ u64 mul_u64_u64_div_u64(u64 a, u64 b, u64 c) ...@@ -190,6 +191,20 @@ u64 mul_u64_u64_div_u64(u64 a, u64 b, u64 c)
/* can a * b overflow ? */ /* can a * b overflow ? */
if (ilog2(a) + ilog2(b) > 62) { if (ilog2(a) + ilog2(b) > 62) {
/*
* Note that the algorithm after the if block below might lose
* some precision and the result is more exact for b > a. So
* exchange a and b if a is bigger than b.
*
* For example with a = 43980465100800, b = 100000000, c = 1000000000
* the below calculation doesn't modify b at all because div == 0
* and then shift becomes 45 + 26 - 62 = 9 and so the result
* becomes 4398035251080. However with a and b swapped the exact
* result is calculated (i.e. 4398046510080).
*/
if (a > b)
swap(a, b);
/* /*
* (b * a) / c is equal to * (b * a) / c is equal to
* *
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment