#include <linux/jiffies.h>
 #include <linux/of.h>
 #include <linux/delay.h>
+#include <linux/util_macros.h>
 
 #include <linux/platform_data/ina2xx.h>
 
  */
 static const int ina226_avg_tab[] = { 1, 4, 16, 64, 128, 256, 512, 1024 };
 
-static int ina226_avg_bits(int avg)
-{
-       int i;
-
-       /* Get the closest average from the tab. */
-       for (i = 0; i < ARRAY_SIZE(ina226_avg_tab) - 1; i++) {
-               if (avg <= (ina226_avg_tab[i] + ina226_avg_tab[i + 1]) / 2)
-                       break;
-       }
-
-       return i; /* Return 0b0111 for values greater than 1024. */
-}
-
 static int ina226_reg_to_interval(u16 config)
 {
        int avg = ina226_avg_tab[INA226_READ_AVG(config)];
 
        avg = DIV_ROUND_CLOSEST(interval * 1000,
                                INA226_TOTAL_CONV_TIME_DEFAULT);
-       avg_bits = ina226_avg_bits(avg);
+       avg_bits = find_closest(avg, ina226_avg_tab,
+                               ARRAY_SIZE(ina226_avg_tab));
 
        return (config & ~INA226_AVG_RD_MASK) | INA226_SHIFT_AVG(avg_bits);
 }