xref: /openbmc/linux/drivers/thermal/imx8mm_thermal.c (revision c900529f3d9161bfde5cca0754f83b4d3c3e0220)
15eed800aSAnson Huang // SPDX-License-Identifier: GPL-2.0
25eed800aSAnson Huang /*
35eed800aSAnson Huang  * Copyright 2020 NXP.
45eed800aSAnson Huang  *
55eed800aSAnson Huang  * Author: Anson Huang <Anson.Huang@nxp.com>
65eed800aSAnson Huang  */
75eed800aSAnson Huang 
82b8f1f03SAnson Huang #include <linux/bitfield.h>
95eed800aSAnson Huang #include <linux/clk.h>
105eed800aSAnson Huang #include <linux/err.h>
115eed800aSAnson Huang #include <linux/io.h>
125eed800aSAnson Huang #include <linux/module.h>
1340329164SMarek Vasut #include <linux/nvmem-consumer.h>
145eed800aSAnson Huang #include <linux/of.h>
155eed800aSAnson Huang #include <linux/platform_device.h>
1640329164SMarek Vasut #include <linux/slab.h>
175eed800aSAnson Huang #include <linux/thermal.h>
185eed800aSAnson Huang 
19de95d134SAlexander Stein #include "thermal_hwmon.h"
205eed800aSAnson Huang 
215eed800aSAnson Huang #define TER			0x0	/* TMU enable */
222b8f1f03SAnson Huang #define TPS			0x4
235eed800aSAnson Huang #define TRITSR			0x20	/* TMU immediate temp */
2440329164SMarek Vasut /* TMU calibration data registers */
2540329164SMarek Vasut #define TASR			0x28
2640329164SMarek Vasut #define TASR_BUF_SLOPE_MASK	GENMASK(19, 16)
2740329164SMarek Vasut #define TASR_BUF_VREF_MASK	GENMASK(4, 0)	/* TMU_V1 */
2840329164SMarek Vasut #define TASR_BUF_VERF_SEL_MASK	GENMASK(1, 0)	/* TMU_V2 */
2940329164SMarek Vasut #define TCALIV(n)		(0x30 + ((n) * 4))
3040329164SMarek Vasut #define TCALIV_EN		BIT(31)
3140329164SMarek Vasut #define TCALIV_HR_MASK		GENMASK(23, 16)	/* TMU_V1 */
3240329164SMarek Vasut #define TCALIV_RT_MASK		GENMASK(7, 0)	/* TMU_V1 */
3340329164SMarek Vasut #define TCALIV_SNSR105C_MASK	GENMASK(27, 16)	/* TMU_V2 */
3440329164SMarek Vasut #define TCALIV_SNSR25C_MASK	GENMASK(11, 0)	/* TMU_V2 */
3540329164SMarek Vasut #define TRIM			0x3c
3640329164SMarek Vasut #define TRIM_BJT_CUR_MASK	GENMASK(23, 20)
3740329164SMarek Vasut #define TRIM_BGR_MASK		GENMASK(31, 28)
3840329164SMarek Vasut #define TRIM_VLSB_MASK		GENMASK(15, 12)
3940329164SMarek Vasut #define TRIM_EN_CH		BIT(7)
405eed800aSAnson Huang 
413de89d88SPaul Gerber #define TER_ADC_PD		BIT(30)
425eed800aSAnson Huang #define TER_EN			BIT(31)
431f455f14SMarcus Folkesson #define TRITSR_TEMP0_VAL_MASK	GENMASK(7, 0)
441f455f14SMarcus Folkesson #define TRITSR_TEMP1_VAL_MASK	GENMASK(23, 16)
455eed800aSAnson Huang 
462b8f1f03SAnson Huang #define PROBE_SEL_ALL		GENMASK(31, 30)
472b8f1f03SAnson Huang 
482b8f1f03SAnson Huang #define probe_status_offset(x)	(30 + x)
492b8f1f03SAnson Huang #define SIGN_BIT		BIT(7)
502b8f1f03SAnson Huang #define TEMP_VAL_MASK		GENMASK(6, 0)
512b8f1f03SAnson Huang 
5240329164SMarek Vasut /* TMU OCOTP calibration data bitfields */
5340329164SMarek Vasut #define ANA0_EN			BIT(25)
5440329164SMarek Vasut #define ANA0_BUF_VREF_MASK	GENMASK(24, 20)
5540329164SMarek Vasut #define ANA0_BUF_SLOPE_MASK	GENMASK(19, 16)
5640329164SMarek Vasut #define ANA0_HR_MASK		GENMASK(15, 8)
5740329164SMarek Vasut #define ANA0_RT_MASK		GENMASK(7, 0)
5840329164SMarek Vasut #define TRIM2_VLSB_MASK		GENMASK(23, 20)
5940329164SMarek Vasut #define TRIM2_BGR_MASK		GENMASK(19, 16)
6040329164SMarek Vasut #define TRIM2_BJT_CUR_MASK	GENMASK(15, 12)
6140329164SMarek Vasut #define TRIM2_BUF_SLOP_SEL_MASK	GENMASK(11, 8)
6240329164SMarek Vasut #define TRIM2_BUF_VERF_SEL_MASK	GENMASK(7, 6)
6340329164SMarek Vasut #define TRIM3_TCA25_0_LSB_MASK	GENMASK(31, 28)
6440329164SMarek Vasut #define TRIM3_TCA40_0_MASK	GENMASK(27, 16)
6540329164SMarek Vasut #define TRIM4_TCA40_1_MASK	GENMASK(31, 20)
6640329164SMarek Vasut #define TRIM4_TCA105_0_MASK	GENMASK(19, 8)
6740329164SMarek Vasut #define TRIM4_TCA25_0_MSB_MASK	GENMASK(7, 0)
6840329164SMarek Vasut #define TRIM5_TCA105_1_MASK	GENMASK(23, 12)
6940329164SMarek Vasut #define TRIM5_TCA25_1_MASK	GENMASK(11, 0)
7040329164SMarek Vasut 
712b8f1f03SAnson Huang #define VER1_TEMP_LOW_LIMIT	10000
722b8f1f03SAnson Huang #define VER2_TEMP_LOW_LIMIT	-40000
732b8f1f03SAnson Huang #define VER2_TEMP_HIGH_LIMIT	125000
742b8f1f03SAnson Huang 
752b8f1f03SAnson Huang #define TMU_VER1		0x1
762b8f1f03SAnson Huang #define TMU_VER2		0x2
772b8f1f03SAnson Huang 
782b8f1f03SAnson Huang struct thermal_soc_data {
792b8f1f03SAnson Huang 	u32 num_sensors;
802b8f1f03SAnson Huang 	u32 version;
812b8f1f03SAnson Huang 	int (*get_temp)(void *, int *);
822b8f1f03SAnson Huang };
832b8f1f03SAnson Huang 
842b8f1f03SAnson Huang struct tmu_sensor {
852b8f1f03SAnson Huang 	struct imx8mm_tmu *priv;
862b8f1f03SAnson Huang 	u32 hw_id;
872b8f1f03SAnson Huang 	struct thermal_zone_device *tzd;
882b8f1f03SAnson Huang };
895eed800aSAnson Huang 
905eed800aSAnson Huang struct imx8mm_tmu {
915eed800aSAnson Huang 	void __iomem *base;
925eed800aSAnson Huang 	struct clk *clk;
932b8f1f03SAnson Huang 	const struct thermal_soc_data *socdata;
94f740e64cSGustavo A. R. Silva 	struct tmu_sensor sensors[];
955eed800aSAnson Huang };
965eed800aSAnson Huang 
imx8mm_tmu_get_temp(void * data,int * temp)972b8f1f03SAnson Huang static int imx8mm_tmu_get_temp(void *data, int *temp)
982b8f1f03SAnson Huang {
992b8f1f03SAnson Huang 	struct tmu_sensor *sensor = data;
1002b8f1f03SAnson Huang 	struct imx8mm_tmu *tmu = sensor->priv;
1012b8f1f03SAnson Huang 	u32 val;
1022b8f1f03SAnson Huang 
1032b8f1f03SAnson Huang 	val = readl_relaxed(tmu->base + TRITSR) & TRITSR_TEMP0_VAL_MASK;
104d37edc73SMarcus Folkesson 
105d37edc73SMarcus Folkesson 	/*
106d37edc73SMarcus Folkesson 	 * Do not validate against the V bit (bit 31) due to errata
107d37edc73SMarcus Folkesson 	 * ERR051272: TMU: Bit 31 of registers TMU_TSCR/TMU_TRITSR/TMU_TRATSR invalid
108d37edc73SMarcus Folkesson 	 */
109d37edc73SMarcus Folkesson 
1102b8f1f03SAnson Huang 	*temp = val * 1000;
111d37edc73SMarcus Folkesson 	if (*temp < VER1_TEMP_LOW_LIMIT || *temp > VER2_TEMP_HIGH_LIMIT)
1122b8f1f03SAnson Huang 		return -EAGAIN;
1132b8f1f03SAnson Huang 
1142b8f1f03SAnson Huang 	return 0;
1152b8f1f03SAnson Huang }
1162b8f1f03SAnson Huang 
imx8mp_tmu_get_temp(void * data,int * temp)1172b8f1f03SAnson Huang static int imx8mp_tmu_get_temp(void *data, int *temp)
1182b8f1f03SAnson Huang {
1192b8f1f03SAnson Huang 	struct tmu_sensor *sensor = data;
1202b8f1f03SAnson Huang 	struct imx8mm_tmu *tmu = sensor->priv;
12176a5c400SAnson Huang 	unsigned long val;
1222b8f1f03SAnson Huang 	bool ready;
1232b8f1f03SAnson Huang 
12476a5c400SAnson Huang 	val = readl_relaxed(tmu->base + TRITSR);
12576a5c400SAnson Huang 	ready = test_bit(probe_status_offset(sensor->hw_id), &val);
1262b8f1f03SAnson Huang 	if (!ready)
1272b8f1f03SAnson Huang 		return -EAGAIN;
1282b8f1f03SAnson Huang 
1292b8f1f03SAnson Huang 	val = sensor->hw_id ? FIELD_GET(TRITSR_TEMP1_VAL_MASK, val) :
1302b8f1f03SAnson Huang 	      FIELD_GET(TRITSR_TEMP0_VAL_MASK, val);
1312b8f1f03SAnson Huang 	if (val & SIGN_BIT) /* negative */
1322b8f1f03SAnson Huang 		val = (~(val & TEMP_VAL_MASK) + 1);
1332b8f1f03SAnson Huang 
1342b8f1f03SAnson Huang 	*temp = val * 1000;
1352b8f1f03SAnson Huang 	if (*temp < VER2_TEMP_LOW_LIMIT || *temp > VER2_TEMP_HIGH_LIMIT)
1362b8f1f03SAnson Huang 		return -EAGAIN;
1372b8f1f03SAnson Huang 
1382b8f1f03SAnson Huang 	return 0;
1392b8f1f03SAnson Huang }
1402b8f1f03SAnson Huang 
tmu_get_temp(struct thermal_zone_device * tz,int * temp)14132fb9a8aSDaniel Lezcano static int tmu_get_temp(struct thermal_zone_device *tz, int *temp)
1425eed800aSAnson Huang {
1435f68d078SDaniel Lezcano 	struct tmu_sensor *sensor = thermal_zone_device_priv(tz);
1442b8f1f03SAnson Huang 	struct imx8mm_tmu *tmu = sensor->priv;
1455eed800aSAnson Huang 
14632fb9a8aSDaniel Lezcano 	return tmu->socdata->get_temp(sensor, temp);
1475eed800aSAnson Huang }
1485eed800aSAnson Huang 
14932fb9a8aSDaniel Lezcano static const struct thermal_zone_device_ops tmu_tz_ops = {
1505eed800aSAnson Huang 	.get_temp = tmu_get_temp,
1515eed800aSAnson Huang };
1525eed800aSAnson Huang 
imx8mm_tmu_enable(struct imx8mm_tmu * tmu,bool enable)1532b8f1f03SAnson Huang static void imx8mm_tmu_enable(struct imx8mm_tmu *tmu, bool enable)
1542b8f1f03SAnson Huang {
1552b8f1f03SAnson Huang 	u32 val;
1562b8f1f03SAnson Huang 
1572b8f1f03SAnson Huang 	val = readl_relaxed(tmu->base + TER);
1582b8f1f03SAnson Huang 	val = enable ? (val | TER_EN) : (val & ~TER_EN);
1593de89d88SPaul Gerber 	if (tmu->socdata->version == TMU_VER2)
1603de89d88SPaul Gerber 		val = enable ? (val & ~TER_ADC_PD) : (val | TER_ADC_PD);
1612b8f1f03SAnson Huang 	writel_relaxed(val, tmu->base + TER);
1622b8f1f03SAnson Huang }
1632b8f1f03SAnson Huang 
imx8mm_tmu_probe_sel_all(struct imx8mm_tmu * tmu)1642b8f1f03SAnson Huang static void imx8mm_tmu_probe_sel_all(struct imx8mm_tmu *tmu)
1652b8f1f03SAnson Huang {
1662b8f1f03SAnson Huang 	u32 val;
1672b8f1f03SAnson Huang 
1682b8f1f03SAnson Huang 	val = readl_relaxed(tmu->base + TPS);
1692b8f1f03SAnson Huang 	val |= PROBE_SEL_ALL;
1702b8f1f03SAnson Huang 	writel_relaxed(val, tmu->base + TPS);
1712b8f1f03SAnson Huang }
1722b8f1f03SAnson Huang 
imx8mm_tmu_probe_set_calib_v1(struct platform_device * pdev,struct imx8mm_tmu * tmu)17340329164SMarek Vasut static int imx8mm_tmu_probe_set_calib_v1(struct platform_device *pdev,
17440329164SMarek Vasut 					 struct imx8mm_tmu *tmu)
17540329164SMarek Vasut {
17640329164SMarek Vasut 	struct device *dev = &pdev->dev;
17740329164SMarek Vasut 	u32 ana0;
17840329164SMarek Vasut 	int ret;
17940329164SMarek Vasut 
18040329164SMarek Vasut 	ret = nvmem_cell_read_u32(&pdev->dev, "calib", &ana0);
181*4afcb58eSAhmad Fatoum 	if (ret)
182*4afcb58eSAhmad Fatoum 		return dev_err_probe(dev, ret, "Failed to read OCOTP nvmem cell\n");
18340329164SMarek Vasut 
18440329164SMarek Vasut 	writel(FIELD_PREP(TASR_BUF_VREF_MASK,
18540329164SMarek Vasut 			  FIELD_GET(ANA0_BUF_VREF_MASK, ana0)) |
18640329164SMarek Vasut 	       FIELD_PREP(TASR_BUF_SLOPE_MASK,
18740329164SMarek Vasut 			  FIELD_GET(ANA0_BUF_SLOPE_MASK, ana0)),
18840329164SMarek Vasut 	       tmu->base + TASR);
18940329164SMarek Vasut 
19040329164SMarek Vasut 	writel(FIELD_PREP(TCALIV_RT_MASK, FIELD_GET(ANA0_RT_MASK, ana0)) |
19140329164SMarek Vasut 	       FIELD_PREP(TCALIV_HR_MASK, FIELD_GET(ANA0_HR_MASK, ana0)) |
19240329164SMarek Vasut 	       ((ana0 & ANA0_EN) ? TCALIV_EN : 0),
19340329164SMarek Vasut 	       tmu->base + TCALIV(0));
19440329164SMarek Vasut 
19540329164SMarek Vasut 	return 0;
19640329164SMarek Vasut }
19740329164SMarek Vasut 
imx8mm_tmu_probe_set_calib_v2(struct platform_device * pdev,struct imx8mm_tmu * tmu)19840329164SMarek Vasut static int imx8mm_tmu_probe_set_calib_v2(struct platform_device *pdev,
19940329164SMarek Vasut 					 struct imx8mm_tmu *tmu)
20040329164SMarek Vasut {
20140329164SMarek Vasut 	struct device *dev = &pdev->dev;
20240329164SMarek Vasut 	struct nvmem_cell *cell;
20340329164SMarek Vasut 	u32 trim[4] = { 0 };
20440329164SMarek Vasut 	size_t len;
20540329164SMarek Vasut 	void *buf;
20640329164SMarek Vasut 
20740329164SMarek Vasut 	cell = nvmem_cell_get(dev, "calib");
20840329164SMarek Vasut 	if (IS_ERR(cell))
20940329164SMarek Vasut 		return PTR_ERR(cell);
21040329164SMarek Vasut 
21140329164SMarek Vasut 	buf = nvmem_cell_read(cell, &len);
21240329164SMarek Vasut 	nvmem_cell_put(cell);
21340329164SMarek Vasut 
21440329164SMarek Vasut 	if (IS_ERR(buf))
21540329164SMarek Vasut 		return PTR_ERR(buf);
21640329164SMarek Vasut 
21740329164SMarek Vasut 	memcpy(trim, buf, min(len, sizeof(trim)));
21840329164SMarek Vasut 	kfree(buf);
21940329164SMarek Vasut 
22040329164SMarek Vasut 	if (len != 16) {
22140329164SMarek Vasut 		dev_err(dev,
22240329164SMarek Vasut 			"OCOTP nvmem cell length is %zu, must be 16.\n", len);
22340329164SMarek Vasut 		return -EINVAL;
22440329164SMarek Vasut 	}
22540329164SMarek Vasut 
22640329164SMarek Vasut 	/* Blank sample hardware */
22740329164SMarek Vasut 	if (!trim[0] && !trim[1] && !trim[2] && !trim[3]) {
22840329164SMarek Vasut 		/* Use a default 25C binary codes */
22940329164SMarek Vasut 		writel(FIELD_PREP(TCALIV_SNSR25C_MASK, 0x63c),
23040329164SMarek Vasut 		       tmu->base + TCALIV(0));
23140329164SMarek Vasut 		writel(FIELD_PREP(TCALIV_SNSR25C_MASK, 0x63c),
23240329164SMarek Vasut 		       tmu->base + TCALIV(1));
23340329164SMarek Vasut 		return 0;
23440329164SMarek Vasut 	}
23540329164SMarek Vasut 
23640329164SMarek Vasut 	writel(FIELD_PREP(TASR_BUF_VERF_SEL_MASK,
23740329164SMarek Vasut 			  FIELD_GET(TRIM2_BUF_VERF_SEL_MASK, trim[0])) |
23840329164SMarek Vasut 	       FIELD_PREP(TASR_BUF_SLOPE_MASK,
23940329164SMarek Vasut 			  FIELD_GET(TRIM2_BUF_SLOP_SEL_MASK, trim[0])),
24040329164SMarek Vasut 	       tmu->base + TASR);
24140329164SMarek Vasut 
24240329164SMarek Vasut 	writel(FIELD_PREP(TRIM_BJT_CUR_MASK,
24340329164SMarek Vasut 			  FIELD_GET(TRIM2_BJT_CUR_MASK, trim[0])) |
24440329164SMarek Vasut 	       FIELD_PREP(TRIM_BGR_MASK, FIELD_GET(TRIM2_BGR_MASK, trim[0])) |
24540329164SMarek Vasut 	       FIELD_PREP(TRIM_VLSB_MASK, FIELD_GET(TRIM2_VLSB_MASK, trim[0])) |
24640329164SMarek Vasut 	       TRIM_EN_CH,
24740329164SMarek Vasut 	       tmu->base + TRIM);
24840329164SMarek Vasut 
24940329164SMarek Vasut 	writel(FIELD_PREP(TCALIV_SNSR25C_MASK,
25040329164SMarek Vasut 			  FIELD_GET(TRIM3_TCA25_0_LSB_MASK, trim[1]) |
25140329164SMarek Vasut 			  (FIELD_GET(TRIM4_TCA25_0_MSB_MASK, trim[2]) << 4)) |
25240329164SMarek Vasut 	       FIELD_PREP(TCALIV_SNSR105C_MASK,
25340329164SMarek Vasut 			  FIELD_GET(TRIM4_TCA105_0_MASK, trim[2])),
25440329164SMarek Vasut 	       tmu->base + TCALIV(0));
25540329164SMarek Vasut 
25640329164SMarek Vasut 	writel(FIELD_PREP(TCALIV_SNSR25C_MASK,
25740329164SMarek Vasut 			  FIELD_GET(TRIM5_TCA25_1_MASK, trim[3])) |
25840329164SMarek Vasut 	       FIELD_PREP(TCALIV_SNSR105C_MASK,
25940329164SMarek Vasut 			  FIELD_GET(TRIM5_TCA105_1_MASK, trim[3])),
26040329164SMarek Vasut 	       tmu->base + TCALIV(1));
26140329164SMarek Vasut 
26240329164SMarek Vasut 	writel(FIELD_PREP(TCALIV_SNSR25C_MASK,
26340329164SMarek Vasut 			  FIELD_GET(TRIM3_TCA40_0_MASK, trim[1])) |
26440329164SMarek Vasut 	       FIELD_PREP(TCALIV_SNSR105C_MASK,
26540329164SMarek Vasut 			  FIELD_GET(TRIM4_TCA40_1_MASK, trim[2])),
26640329164SMarek Vasut 	       tmu->base + TCALIV(2));
26740329164SMarek Vasut 
26840329164SMarek Vasut 	return 0;
26940329164SMarek Vasut }
27040329164SMarek Vasut 
imx8mm_tmu_probe_set_calib(struct platform_device * pdev,struct imx8mm_tmu * tmu)27140329164SMarek Vasut static int imx8mm_tmu_probe_set_calib(struct platform_device *pdev,
27240329164SMarek Vasut 				      struct imx8mm_tmu *tmu)
27340329164SMarek Vasut {
27440329164SMarek Vasut 	struct device *dev = &pdev->dev;
27540329164SMarek Vasut 
27640329164SMarek Vasut 	/*
27740329164SMarek Vasut 	 * Lack of calibration data OCOTP reference is not considered
27840329164SMarek Vasut 	 * fatal to retain compatibility with old DTs. It is however
27940329164SMarek Vasut 	 * strongly recommended to update such old DTs to get correct
28040329164SMarek Vasut 	 * temperature compensation values for each SoC.
28140329164SMarek Vasut 	 */
28286df7d19SRob Herring 	if (!of_property_present(pdev->dev.of_node, "nvmem-cells")) {
28340329164SMarek Vasut 		dev_warn(dev,
28440329164SMarek Vasut 			 "No OCOTP nvmem reference found, SoC-specific calibration not loaded. Please update your DT.\n");
28540329164SMarek Vasut 		return 0;
28640329164SMarek Vasut 	}
28740329164SMarek Vasut 
28840329164SMarek Vasut 	if (tmu->socdata->version == TMU_VER1)
28940329164SMarek Vasut 		return imx8mm_tmu_probe_set_calib_v1(pdev, tmu);
29040329164SMarek Vasut 
29140329164SMarek Vasut 	return imx8mm_tmu_probe_set_calib_v2(pdev, tmu);
29240329164SMarek Vasut }
29340329164SMarek Vasut 
imx8mm_tmu_probe(struct platform_device * pdev)2945eed800aSAnson Huang static int imx8mm_tmu_probe(struct platform_device *pdev)
2955eed800aSAnson Huang {
2962b8f1f03SAnson Huang 	const struct thermal_soc_data *data;
2975eed800aSAnson Huang 	struct imx8mm_tmu *tmu;
2985eed800aSAnson Huang 	int ret;
2992b8f1f03SAnson Huang 	int i;
3005eed800aSAnson Huang 
3012b8f1f03SAnson Huang 	data = of_device_get_match_data(&pdev->dev);
3022b8f1f03SAnson Huang 
3032b8f1f03SAnson Huang 	tmu = devm_kzalloc(&pdev->dev, struct_size(tmu, sensors,
3042b8f1f03SAnson Huang 			   data->num_sensors), GFP_KERNEL);
3055eed800aSAnson Huang 	if (!tmu)
3065eed800aSAnson Huang 		return -ENOMEM;
3075eed800aSAnson Huang 
3082b8f1f03SAnson Huang 	tmu->socdata = data;
3092b8f1f03SAnson Huang 
3105eed800aSAnson Huang 	tmu->base = devm_platform_ioremap_resource(pdev, 0);
3115eed800aSAnson Huang 	if (IS_ERR(tmu->base))
3125eed800aSAnson Huang 		return PTR_ERR(tmu->base);
3135eed800aSAnson Huang 
3145eed800aSAnson Huang 	tmu->clk = devm_clk_get(&pdev->dev, NULL);
3158790710aSAnson Huang 	if (IS_ERR(tmu->clk))
3168790710aSAnson Huang 		return dev_err_probe(&pdev->dev, PTR_ERR(tmu->clk),
3178790710aSAnson Huang 				     "failed to get tmu clock\n");
3185eed800aSAnson Huang 
3195eed800aSAnson Huang 	ret = clk_prepare_enable(tmu->clk);
3205eed800aSAnson Huang 	if (ret) {
3215eed800aSAnson Huang 		dev_err(&pdev->dev, "failed to enable tmu clock: %d\n", ret);
3225eed800aSAnson Huang 		return ret;
3235eed800aSAnson Huang 	}
3245eed800aSAnson Huang 
3252b8f1f03SAnson Huang 	/* disable the monitor during initialization */
3262b8f1f03SAnson Huang 	imx8mm_tmu_enable(tmu, false);
3272b8f1f03SAnson Huang 
3282b8f1f03SAnson Huang 	for (i = 0; i < data->num_sensors; i++) {
3292b8f1f03SAnson Huang 		tmu->sensors[i].priv = tmu;
3302b8f1f03SAnson Huang 		tmu->sensors[i].tzd =
33132fb9a8aSDaniel Lezcano 			devm_thermal_of_zone_register(&pdev->dev, i,
3322b8f1f03SAnson Huang 						      &tmu->sensors[i],
3332b8f1f03SAnson Huang 						      &tmu_tz_ops);
3342b8f1f03SAnson Huang 		if (IS_ERR(tmu->sensors[i].tzd)) {
335ce662ccdSFabio Estevam 			ret = PTR_ERR(tmu->sensors[i].tzd);
3365eed800aSAnson Huang 			dev_err(&pdev->dev,
3372b8f1f03SAnson Huang 				"failed to register thermal zone sensor[%d]: %d\n",
3382b8f1f03SAnson Huang 				i, ret);
339e57eb8b5SFabio Estevam 			goto disable_clk;
3402b8f1f03SAnson Huang 		}
3412b8f1f03SAnson Huang 		tmu->sensors[i].hw_id = i;
342de95d134SAlexander Stein 
343b0526e02SYangtao Li 		devm_thermal_add_hwmon_sysfs(&pdev->dev, tmu->sensors[i].tzd);
3445eed800aSAnson Huang 	}
3455eed800aSAnson Huang 
3465eed800aSAnson Huang 	platform_set_drvdata(pdev, tmu);
3475eed800aSAnson Huang 
34840329164SMarek Vasut 	ret = imx8mm_tmu_probe_set_calib(pdev, tmu);
34940329164SMarek Vasut 	if (ret)
35040329164SMarek Vasut 		goto disable_clk;
35140329164SMarek Vasut 
3522b8f1f03SAnson Huang 	/* enable all the probes for V2 TMU */
3532b8f1f03SAnson Huang 	if (tmu->socdata->version == TMU_VER2)
3542b8f1f03SAnson Huang 		imx8mm_tmu_probe_sel_all(tmu);
3552b8f1f03SAnson Huang 
3565eed800aSAnson Huang 	/* enable the monitor */
3572b8f1f03SAnson Huang 	imx8mm_tmu_enable(tmu, true);
3585eed800aSAnson Huang 
3595eed800aSAnson Huang 	return 0;
360e57eb8b5SFabio Estevam 
361e57eb8b5SFabio Estevam disable_clk:
362e57eb8b5SFabio Estevam 	clk_disable_unprepare(tmu->clk);
363e57eb8b5SFabio Estevam 	return ret;
3645eed800aSAnson Huang }
3655eed800aSAnson Huang 
imx8mm_tmu_remove(struct platform_device * pdev)3665eed800aSAnson Huang static int imx8mm_tmu_remove(struct platform_device *pdev)
3675eed800aSAnson Huang {
3685eed800aSAnson Huang 	struct imx8mm_tmu *tmu = platform_get_drvdata(pdev);
3695eed800aSAnson Huang 
3705eed800aSAnson Huang 	/* disable TMU */
3712b8f1f03SAnson Huang 	imx8mm_tmu_enable(tmu, false);
3725eed800aSAnson Huang 
3735eed800aSAnson Huang 	clk_disable_unprepare(tmu->clk);
3745eed800aSAnson Huang 	platform_set_drvdata(pdev, NULL);
3755eed800aSAnson Huang 
3765eed800aSAnson Huang 	return 0;
3775eed800aSAnson Huang }
3785eed800aSAnson Huang 
3792b8f1f03SAnson Huang static struct thermal_soc_data imx8mm_tmu_data = {
3802b8f1f03SAnson Huang 	.num_sensors = 1,
3812b8f1f03SAnson Huang 	.version = TMU_VER1,
3822b8f1f03SAnson Huang 	.get_temp = imx8mm_tmu_get_temp,
3832b8f1f03SAnson Huang };
3842b8f1f03SAnson Huang 
3852b8f1f03SAnson Huang static struct thermal_soc_data imx8mp_tmu_data = {
3862b8f1f03SAnson Huang 	.num_sensors = 2,
3872b8f1f03SAnson Huang 	.version = TMU_VER2,
3882b8f1f03SAnson Huang 	.get_temp = imx8mp_tmu_get_temp,
3892b8f1f03SAnson Huang };
3902b8f1f03SAnson Huang 
3915eed800aSAnson Huang static const struct of_device_id imx8mm_tmu_table[] = {
3922b8f1f03SAnson Huang 	{ .compatible = "fsl,imx8mm-tmu", .data = &imx8mm_tmu_data, },
3932b8f1f03SAnson Huang 	{ .compatible = "fsl,imx8mp-tmu", .data = &imx8mp_tmu_data, },
3945eed800aSAnson Huang 	{ },
3955eed800aSAnson Huang };
3964b9e373eSAnson Huang MODULE_DEVICE_TABLE(of, imx8mm_tmu_table);
3975eed800aSAnson Huang 
3985eed800aSAnson Huang static struct platform_driver imx8mm_tmu = {
3995eed800aSAnson Huang 	.driver = {
4005eed800aSAnson Huang 		.name	= "i.mx8mm_thermal",
4015eed800aSAnson Huang 		.of_match_table = imx8mm_tmu_table,
4025eed800aSAnson Huang 	},
4035eed800aSAnson Huang 	.probe = imx8mm_tmu_probe,
4045eed800aSAnson Huang 	.remove = imx8mm_tmu_remove,
4055eed800aSAnson Huang };
4065eed800aSAnson Huang module_platform_driver(imx8mm_tmu);
4075eed800aSAnson Huang 
4085eed800aSAnson Huang MODULE_AUTHOR("Anson Huang <Anson.Huang@nxp.com>");
4095eed800aSAnson Huang MODULE_DESCRIPTION("i.MX8MM Thermal Monitor Unit driver");
4105eed800aSAnson Huang MODULE_LICENSE("GPL v2");
411