1 files changed, 7 insertions, 2 deletions

diff --git a/drivers/sh/clk/core.c b/drivers/sh/clk/core.c
index db257a35e71a..7715de2629c1 100644
--- a/drivers/sh/clk/core.c
+++ b/drivers/sh/clk/core.c
@@ -355,7 +355,7 @@ static int clk_establish_mapping(struct clk *clk)
                  */
                 if (!clk->parent) {
                         clk->mapping = &dummy_mapping;
-                        return 0;
+                        goto out;
                 }
 
                 /*
@@ -384,6 +384,9 @@ static int clk_establish_mapping(struct clk *clk)
         }
 
         clk->mapping = mapping;
+out:
+        clk->mapped_reg = clk->mapping->base;
+        clk->mapped_reg += (phys_addr_t)clk->enable_reg - clk->mapping->phys;
         return 0;
 }
 
@@ -402,10 +405,12 @@ static void clk_teardown_mapping(struct clk *clk)
 
         /* Nothing to do */
         if (mapping == &dummy_mapping)
-                return;
+                goto out;
 
         kref_put(&mapping->ref, clk_destroy_mapping);
         clk->mapping = NULL;
+out:
+        clk->mapped_reg = NULL;
 }
 
 int clk_register(struct clk *clk)