#include "internal.h"
 
+static inline unsigned int regcache_flat_get_index(const struct regmap *map,
+                                                  unsigned int reg)
+{
+       return regcache_get_index_by_order(map, reg);
+}
+
 static int regcache_flat_init(struct regmap *map)
 {
        int i;
        unsigned int *cache;
 
-       map->cache = kcalloc(map->max_register + 1, sizeof(unsigned int),
-                            GFP_KERNEL);
+       if (!map || map->reg_stride_order < 0)
+               return -EINVAL;
+
+       map->cache = kcalloc(regcache_flat_get_index(map, map->max_register)
+                            + 1, sizeof(unsigned int), GFP_KERNEL);
        if (!map->cache)
                return -ENOMEM;
 
        cache = map->cache;
 
        for (i = 0; i < map->num_reg_defaults; i++)
-               cache[map->reg_defaults[i].reg] = map->reg_defaults[i].def;
+               cache[regcache_flat_get_index(map, map->reg_defaults[i].reg)] =
+                               map->reg_defaults[i].def;
 
        return 0;
 }
 {
        unsigned int *cache = map->cache;
 
-       *value = cache[reg];
+       *value = cache[regcache_flat_get_index(map, reg)];
 
        return 0;
 }
 {
        unsigned int *cache = map->cache;
 
-       cache[reg] = value;
+       cache[regcache_flat_get_index(map, reg)] = value;
 
        return 0;
 }