diff --git a/drivers/soc/rockchip/rockchip_ipa.c b/drivers/soc/rockchip/rockchip_ipa.c
index 37ee0dbe1e5b..5c5be82aefeb 100644
--- a/drivers/soc/rockchip/rockchip_ipa.c
+++ b/drivers/soc/rockchip/rockchip_ipa.c
@@ -6,11 +6,12 @@
 #include <linux/of.h>
 #include <linux/thermal.h>
 #include <soc/rockchip/rockchip_ipa.h>
+#include <soc/rockchip/rockchip_opp_select.h>
 #include <trace/events/thermal.h>
 
 #define FALLBACK_STATIC_TEMPERATURE 55000
 
-int rockchip_ipa_power_model_init(struct device *dev,
+int rockchip_ipa_power_model_init(struct device *dev, char *lkg_name,
 				  struct ipa_power_model_data **data)
 {
 	struct device_node *model_node;
@@ -57,6 +58,7 @@ int rockchip_ipa_power_model_init(struct device *dev,
 		ret = -EINVAL;
 		goto err;
 	}
+	rockchip_of_get_leakage(dev, lkg_name, &model_data->leakage);
 	of_property_read_u32(model_node, "ref-leakage",
 			     &model_data->ref_leakage);
 	*data = model_data;
diff --git a/include/soc/rockchip/rockchip_ipa.h b/include/soc/rockchip/rockchip_ipa.h
index 9525f45a17c1..a8589c023868 100644
--- a/include/soc/rockchip/rockchip_ipa.h
+++ b/include/soc/rockchip/rockchip_ipa.h
@@ -15,14 +15,14 @@ struct ipa_power_model_data {
 };
 
 #ifdef CONFIG_ROCKCHIP_IPA
-int rockchip_ipa_power_model_init(struct device *dev,
+int rockchip_ipa_power_model_init(struct device *dev, char *lkg_name,
 				  struct ipa_power_model_data **data);
 unsigned long
 rockchip_ipa_get_static_power(struct ipa_power_model_data *model_data,
 			      unsigned long voltage_mv);
 #else
 static inline int
-rockchip_ipa_power_model_init(struct device *dev,
+rockchip_ipa_power_model_init(struct device *dev, char *lkg_name,
 			      struct ipa_power_model_data **data)
 {
 	return -ENOTSUPP;