1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/thermal/samsung/exynos_tmu.c b/drivers/thermal/samsung/exynos_tmu.c
index 33ee527fd211..e4d75a1a17ae 100644
--- a/drivers/thermal/samsung/exynos_tmu.c
+++ b/drivers/thermal/samsung/exynos_tmu.c
@@ -161,7 +161,9 @@ static int exynos_tmu_initialize(struct platform_device *pdev)
                 }
         }
 
-        if (TMU_SUPPORTS(pdata, TRIM_RELOAD)) {
+        if (data->soc == SOC_ARCH_EXYNOS3250 ||
+            data->soc == SOC_ARCH_EXYNOS4412 ||
+            data->soc == SOC_ARCH_EXYNOS5250) {
                 if (data->soc == SOC_ARCH_EXYNOS3250) {
                         ctrl = readl(data->base + EXYNOS_TMU_TRIMINFO_CON1);
                         ctrl |= EXYNOS_TRIMINFO_RELOAD_ENABLE;
@@ -626,6 +628,7 @@ static int exynos_tmu_probe(struct platform_device *pdev)
             pdata->type == SOC_ARCH_EXYNOS4412 ||
             pdata->type == SOC_ARCH_EXYNOS5250 ||
             pdata->type == SOC_ARCH_EXYNOS5260 ||
+            pdata->type == SOC_ARCH_EXYNOS5420 ||
             pdata->type == SOC_ARCH_EXYNOS5420_TRIMINFO ||
             pdata->type == SOC_ARCH_EXYNOS5440)
                 data->soc = pdata->type;