| /* | 
 |  * HiSilicon thermal sensor driver | 
 |  * | 
 |  * Copyright (c) 2014-2015 HiSilicon Limited. | 
 |  * Copyright (c) 2014-2015 Linaro Limited. | 
 |  * | 
 |  * Xinwei Kong <kong.kongxinwei@hisilicon.com> | 
 |  * Leo Yan <leo.yan@linaro.org> | 
 |  * | 
 |  * This program is free software; you can redistribute it and/or modify | 
 |  * it under the terms of the GNU General Public License version 2 as | 
 |  * published by the Free Software Foundation. | 
 |  * | 
 |  * This program is distributed "as is" WITHOUT ANY WARRANTY of any | 
 |  * kind, whether express or implied; without even the implied warranty | 
 |  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 
 |  * GNU General Public License for more details. | 
 |  */ | 
 |  | 
 | #include <linux/cpufreq.h> | 
 | #include <linux/delay.h> | 
 | #include <linux/interrupt.h> | 
 | #include <linux/module.h> | 
 | #include <linux/platform_device.h> | 
 | #include <linux/io.h> | 
 | #include <linux/of_device.h> | 
 |  | 
 | #include "thermal_core.h" | 
 |  | 
 | #define HI6220_TEMP0_LAG			(0x0) | 
 | #define HI6220_TEMP0_TH				(0x4) | 
 | #define HI6220_TEMP0_RST_TH			(0x8) | 
 | #define HI6220_TEMP0_CFG			(0xC) | 
 | #define HI6220_TEMP0_CFG_SS_MSK			(0xF000) | 
 | #define HI6220_TEMP0_CFG_HDAK_MSK		(0x30) | 
 | #define HI6220_TEMP0_EN				(0x10) | 
 | #define HI6220_TEMP0_INT_EN			(0x14) | 
 | #define HI6220_TEMP0_INT_CLR			(0x18) | 
 | #define HI6220_TEMP0_RST_MSK			(0x1C) | 
 | #define HI6220_TEMP0_VALUE			(0x28) | 
 |  | 
 | #define HI3660_OFFSET(chan)		((chan) * 0x40) | 
 | #define HI3660_TEMP(chan)		(HI3660_OFFSET(chan) + 0x1C) | 
 | #define HI3660_TH(chan)			(HI3660_OFFSET(chan) + 0x20) | 
 | #define HI3660_LAG(chan)		(HI3660_OFFSET(chan) + 0x28) | 
 | #define HI3660_INT_EN(chan)		(HI3660_OFFSET(chan) + 0x2C) | 
 | #define HI3660_INT_CLR(chan)		(HI3660_OFFSET(chan) + 0x30) | 
 |  | 
 | #define HI6220_TEMP_BASE			(-60000) | 
 | #define HI6220_TEMP_RESET			(100000) | 
 | #define HI6220_TEMP_STEP			(785) | 
 | #define HI6220_TEMP_LAG				(3500) | 
 |  | 
 | #define HI3660_TEMP_BASE		(-63780) | 
 | #define HI3660_TEMP_STEP		(205) | 
 | #define HI3660_TEMP_LAG			(4000) | 
 |  | 
 | #define HI6220_CLUSTER0_SENSOR		2 | 
 | #define HI6220_CLUSTER1_SENSOR		1 | 
 |  | 
 | #define HI3660_LITTLE_SENSOR		0 | 
 | #define HI3660_BIG_SENSOR		1 | 
 | #define HI3660_G3D_SENSOR		2 | 
 | #define HI3660_MODEM_SENSOR		3 | 
 |  | 
 | struct hisi_thermal_data; | 
 |  | 
 | struct hisi_thermal_sensor { | 
 | 	struct hisi_thermal_data *data; | 
 | 	struct thermal_zone_device *tzd; | 
 | 	const char *irq_name; | 
 | 	uint32_t id; | 
 | 	uint32_t thres_temp; | 
 | }; | 
 |  | 
 | struct hisi_thermal_ops { | 
 | 	int (*get_temp)(struct hisi_thermal_sensor *sensor); | 
 | 	int (*enable_sensor)(struct hisi_thermal_sensor *sensor); | 
 | 	int (*disable_sensor)(struct hisi_thermal_sensor *sensor); | 
 | 	int (*irq_handler)(struct hisi_thermal_sensor *sensor); | 
 | 	int (*probe)(struct hisi_thermal_data *data); | 
 | }; | 
 |  | 
 | struct hisi_thermal_data { | 
 | 	const struct hisi_thermal_ops *ops; | 
 | 	struct hisi_thermal_sensor *sensor; | 
 | 	struct platform_device *pdev; | 
 | 	struct clk *clk; | 
 | 	void __iomem *regs; | 
 | 	int nr_sensors; | 
 | }; | 
 |  | 
 | /* | 
 |  * The temperature computation on the tsensor is as follow: | 
 |  *	Unit: millidegree Celsius | 
 |  *	Step: 200/255 (0.7843) | 
 |  *	Temperature base: -60°C | 
 |  * | 
 |  * The register is programmed in temperature steps, every step is 785 | 
 |  * millidegree and begins at -60 000 m°C | 
 |  * | 
 |  * The temperature from the steps: | 
 |  * | 
 |  *	Temp = TempBase + (steps x 785) | 
 |  * | 
 |  * and the steps from the temperature: | 
 |  * | 
 |  *	steps = (Temp - TempBase) / 785 | 
 |  * | 
 |  */ | 
 | static inline int hi6220_thermal_step_to_temp(int step) | 
 | { | 
 | 	return HI6220_TEMP_BASE + (step * HI6220_TEMP_STEP); | 
 | } | 
 |  | 
 | static inline int hi6220_thermal_temp_to_step(int temp) | 
 | { | 
 | 	return DIV_ROUND_UP(temp - HI6220_TEMP_BASE, HI6220_TEMP_STEP); | 
 | } | 
 |  | 
 | /* | 
 |  * for Hi3660, | 
 |  *	Step: 189/922 (0.205) | 
 |  *	Temperature base: -63.780°C | 
 |  * | 
 |  * The register is programmed in temperature steps, every step is 205 | 
 |  * millidegree and begins at -63 780 m°C | 
 |  */ | 
 | static inline int hi3660_thermal_step_to_temp(int step) | 
 | { | 
 | 	return HI3660_TEMP_BASE + step * HI3660_TEMP_STEP; | 
 | } | 
 |  | 
 | static inline int hi3660_thermal_temp_to_step(int temp) | 
 | { | 
 | 	return DIV_ROUND_UP(temp - HI3660_TEMP_BASE, HI3660_TEMP_STEP); | 
 | } | 
 |  | 
 | /* | 
 |  * The lag register contains 5 bits encoding the temperature in steps. | 
 |  * | 
 |  * Each time the temperature crosses the threshold boundary, an | 
 |  * interrupt is raised. It could be when the temperature is going | 
 |  * above the threshold or below. However, if the temperature is | 
 |  * fluctuating around this value due to the load, we can receive | 
 |  * several interrupts which may not desired. | 
 |  * | 
 |  * We can setup a temperature representing the delta between the | 
 |  * threshold and the current temperature when the temperature is | 
 |  * decreasing. | 
 |  * | 
 |  * For instance: the lag register is 5°C, the threshold is 65°C, when | 
 |  * the temperature reaches 65°C an interrupt is raised and when the | 
 |  * temperature decrease to 65°C - 5°C another interrupt is raised. | 
 |  * | 
 |  * A very short lag can lead to an interrupt storm, a long lag | 
 |  * increase the latency to react to the temperature changes.  In our | 
 |  * case, that is not really a problem as we are polling the | 
 |  * temperature. | 
 |  * | 
 |  * [0:4] : lag register | 
 |  * | 
 |  * The temperature is coded in steps, cf. HI6220_TEMP_STEP. | 
 |  * | 
 |  * Min : 0x00 :  0.0 °C | 
 |  * Max : 0x1F : 24.3 °C | 
 |  * | 
 |  * The 'value' parameter is in milliCelsius. | 
 |  */ | 
 | static inline void hi6220_thermal_set_lag(void __iomem *addr, int value) | 
 | { | 
 | 	writel(DIV_ROUND_UP(value, HI6220_TEMP_STEP) & 0x1F, | 
 | 			addr + HI6220_TEMP0_LAG); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_alarm_clear(void __iomem *addr, int value) | 
 | { | 
 | 	writel(value, addr + HI6220_TEMP0_INT_CLR); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_alarm_enable(void __iomem *addr, int value) | 
 | { | 
 | 	writel(value, addr + HI6220_TEMP0_INT_EN); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_alarm_set(void __iomem *addr, int temp) | 
 | { | 
 | 	writel(hi6220_thermal_temp_to_step(temp) | 0x0FFFFFF00, | 
 | 	       addr + HI6220_TEMP0_TH); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_reset_set(void __iomem *addr, int temp) | 
 | { | 
 | 	writel(hi6220_thermal_temp_to_step(temp), addr + HI6220_TEMP0_RST_TH); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_reset_enable(void __iomem *addr, int value) | 
 | { | 
 | 	writel(value, addr + HI6220_TEMP0_RST_MSK); | 
 | } | 
 |  | 
 | static inline void hi6220_thermal_enable(void __iomem *addr, int value) | 
 | { | 
 | 	writel(value, addr + HI6220_TEMP0_EN); | 
 | } | 
 |  | 
 | static inline int hi6220_thermal_get_temperature(void __iomem *addr) | 
 | { | 
 | 	return hi6220_thermal_step_to_temp(readl(addr + HI6220_TEMP0_VALUE)); | 
 | } | 
 |  | 
 | /* | 
 |  * [0:6] lag register | 
 |  * | 
 |  * The temperature is coded in steps, cf. HI3660_TEMP_STEP. | 
 |  * | 
 |  * Min : 0x00 :  0.0 °C | 
 |  * Max : 0x7F : 26.0 °C | 
 |  * | 
 |  */ | 
 | static inline void hi3660_thermal_set_lag(void __iomem *addr, | 
 | 					  int id, int value) | 
 | { | 
 | 	writel(DIV_ROUND_UP(value, HI3660_TEMP_STEP) & 0x7F, | 
 | 			addr + HI3660_LAG(id)); | 
 | } | 
 |  | 
 | static inline void hi3660_thermal_alarm_clear(void __iomem *addr, | 
 | 					      int id, int value) | 
 | { | 
 | 	writel(value, addr + HI3660_INT_CLR(id)); | 
 | } | 
 |  | 
 | static inline void hi3660_thermal_alarm_enable(void __iomem *addr, | 
 | 					       int id, int value) | 
 | { | 
 | 	writel(value, addr + HI3660_INT_EN(id)); | 
 | } | 
 |  | 
 | static inline void hi3660_thermal_alarm_set(void __iomem *addr, | 
 | 					    int id, int value) | 
 | { | 
 | 	writel(value, addr + HI3660_TH(id)); | 
 | } | 
 |  | 
 | static inline int hi3660_thermal_get_temperature(void __iomem *addr, int id) | 
 | { | 
 | 	return hi3660_thermal_step_to_temp(readl(addr + HI3660_TEMP(id))); | 
 | } | 
 |  | 
 | /* | 
 |  * Temperature configuration register - Sensor selection | 
 |  * | 
 |  * Bits [19:12] | 
 |  * | 
 |  * 0x0: local sensor (default) | 
 |  * 0x1: remote sensor 1 (ACPU cluster 1) | 
 |  * 0x2: remote sensor 2 (ACPU cluster 0) | 
 |  * 0x3: remote sensor 3 (G3D) | 
 |  */ | 
 | static inline void hi6220_thermal_sensor_select(void __iomem *addr, int sensor) | 
 | { | 
 | 	writel((readl(addr + HI6220_TEMP0_CFG) & ~HI6220_TEMP0_CFG_SS_MSK) | | 
 | 	       (sensor << 12), addr + HI6220_TEMP0_CFG); | 
 | } | 
 |  | 
 | /* | 
 |  * Temperature configuration register - Hdak conversion polling interval | 
 |  * | 
 |  * Bits [5:4] | 
 |  * | 
 |  * 0x0 :   0.768 ms | 
 |  * 0x1 :   6.144 ms | 
 |  * 0x2 :  49.152 ms | 
 |  * 0x3 : 393.216 ms | 
 |  */ | 
 | static inline void hi6220_thermal_hdak_set(void __iomem *addr, int value) | 
 | { | 
 | 	writel((readl(addr + HI6220_TEMP0_CFG) & ~HI6220_TEMP0_CFG_HDAK_MSK) | | 
 | 	       (value << 4), addr + HI6220_TEMP0_CFG); | 
 | } | 
 |  | 
 | static int hi6220_thermal_irq_handler(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	hi6220_thermal_alarm_clear(data->regs, 1); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi3660_thermal_irq_handler(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	hi3660_thermal_alarm_clear(data->regs, sensor->id, 1); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi6220_thermal_get_temp(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	return hi6220_thermal_get_temperature(data->regs); | 
 | } | 
 |  | 
 | static int hi3660_thermal_get_temp(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	return hi3660_thermal_get_temperature(data->regs, sensor->id); | 
 | } | 
 |  | 
 | static int hi6220_thermal_disable_sensor(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	/* disable sensor module */ | 
 | 	hi6220_thermal_enable(data->regs, 0); | 
 | 	hi6220_thermal_alarm_enable(data->regs, 0); | 
 | 	hi6220_thermal_reset_enable(data->regs, 0); | 
 |  | 
 | 	clk_disable_unprepare(data->clk); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi3660_thermal_disable_sensor(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	/* disable sensor module */ | 
 | 	hi3660_thermal_alarm_enable(data->regs, sensor->id, 0); | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi6220_thermal_enable_sensor(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 | 	int ret; | 
 |  | 
 | 	/* enable clock for tsensor */ | 
 | 	ret = clk_prepare_enable(data->clk); | 
 | 	if (ret) | 
 | 		return ret; | 
 |  | 
 | 	/* disable module firstly */ | 
 | 	hi6220_thermal_reset_enable(data->regs, 0); | 
 | 	hi6220_thermal_enable(data->regs, 0); | 
 |  | 
 | 	/* select sensor id */ | 
 | 	hi6220_thermal_sensor_select(data->regs, sensor->id); | 
 |  | 
 | 	/* setting the hdak time */ | 
 | 	hi6220_thermal_hdak_set(data->regs, 0); | 
 |  | 
 | 	/* setting lag value between current temp and the threshold */ | 
 | 	hi6220_thermal_set_lag(data->regs, HI6220_TEMP_LAG); | 
 |  | 
 | 	/* enable for interrupt */ | 
 | 	hi6220_thermal_alarm_set(data->regs, sensor->thres_temp); | 
 |  | 
 | 	hi6220_thermal_reset_set(data->regs, HI6220_TEMP_RESET); | 
 |  | 
 | 	/* enable module */ | 
 | 	hi6220_thermal_reset_enable(data->regs, 1); | 
 | 	hi6220_thermal_enable(data->regs, 1); | 
 |  | 
 | 	hi6220_thermal_alarm_clear(data->regs, 0); | 
 | 	hi6220_thermal_alarm_enable(data->regs, 1); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi3660_thermal_enable_sensor(struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	unsigned int value; | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	/* disable interrupt */ | 
 | 	hi3660_thermal_alarm_enable(data->regs, sensor->id, 0); | 
 |  | 
 | 	/* setting lag value between current temp and the threshold */ | 
 | 	hi3660_thermal_set_lag(data->regs, sensor->id, HI3660_TEMP_LAG); | 
 |  | 
 | 	/* set interrupt threshold */ | 
 | 	value = hi3660_thermal_temp_to_step(sensor->thres_temp); | 
 | 	hi3660_thermal_alarm_set(data->regs, sensor->id, value); | 
 |  | 
 | 	/* enable interrupt */ | 
 | 	hi3660_thermal_alarm_clear(data->regs, sensor->id, 1); | 
 | 	hi3660_thermal_alarm_enable(data->regs, sensor->id, 1); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi6220_thermal_probe(struct hisi_thermal_data *data) | 
 | { | 
 | 	struct platform_device *pdev = data->pdev; | 
 | 	struct device *dev = &pdev->dev; | 
 | 	int ret; | 
 |  | 
 | 	data->clk = devm_clk_get(dev, "thermal_clk"); | 
 | 	if (IS_ERR(data->clk)) { | 
 | 		ret = PTR_ERR(data->clk); | 
 | 		if (ret != -EPROBE_DEFER) | 
 | 			dev_err(dev, "failed to get thermal clk: %d\n", ret); | 
 | 		return ret; | 
 | 	} | 
 |  | 
 | 	data->sensor = devm_kzalloc(dev, sizeof(*data->sensor), GFP_KERNEL); | 
 | 	if (!data->sensor) | 
 | 		return -ENOMEM; | 
 |  | 
 | 	data->sensor[0].id = HI6220_CLUSTER0_SENSOR; | 
 | 	data->sensor[0].irq_name = "tsensor_intr"; | 
 | 	data->sensor[0].data = data; | 
 | 	data->nr_sensors = 1; | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hi3660_thermal_probe(struct hisi_thermal_data *data) | 
 | { | 
 | 	struct platform_device *pdev = data->pdev; | 
 | 	struct device *dev = &pdev->dev; | 
 |  | 
 | 	data->nr_sensors = 1; | 
 |  | 
 | 	data->sensor = devm_kzalloc(dev, sizeof(*data->sensor) * | 
 | 				    data->nr_sensors, GFP_KERNEL); | 
 | 	if (!data->sensor) | 
 | 		return -ENOMEM; | 
 |  | 
 | 	data->sensor[0].id = HI3660_BIG_SENSOR; | 
 | 	data->sensor[0].irq_name = "tsensor_a73"; | 
 | 	data->sensor[0].data = data; | 
 |  | 
 | 	data->sensor[1].id = HI3660_LITTLE_SENSOR; | 
 | 	data->sensor[1].irq_name = "tsensor_a53"; | 
 | 	data->sensor[1].data = data; | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hisi_thermal_get_temp(void *__data, int *temp) | 
 | { | 
 | 	struct hisi_thermal_sensor *sensor = __data; | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 |  | 
 | 	*temp = data->ops->get_temp(sensor); | 
 |  | 
 | 	dev_dbg(&data->pdev->dev, "tzd=%p, id=%d, temp=%d, thres=%d\n", | 
 | 		sensor->tzd, sensor->id, *temp, sensor->thres_temp); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static const struct thermal_zone_of_device_ops hisi_of_thermal_ops = { | 
 | 	.get_temp = hisi_thermal_get_temp, | 
 | }; | 
 |  | 
 | static irqreturn_t hisi_thermal_alarm_irq_thread(int irq, void *dev) | 
 | { | 
 | 	struct hisi_thermal_sensor *sensor = dev; | 
 | 	struct hisi_thermal_data *data = sensor->data; | 
 | 	int temp = 0; | 
 |  | 
 | 	data->ops->irq_handler(sensor); | 
 |  | 
 | 	hisi_thermal_get_temp(sensor, &temp); | 
 |  | 
 | 	if (temp >= sensor->thres_temp) { | 
 | 		dev_crit(&data->pdev->dev, | 
 | 			 "sensor <%d> THERMAL ALARM: %d > %d\n", | 
 | 			 sensor->id, temp, sensor->thres_temp); | 
 |  | 
 | 		thermal_zone_device_update(sensor->tzd, | 
 | 					   THERMAL_EVENT_UNSPECIFIED); | 
 |  | 
 | 	} else { | 
 | 		dev_crit(&data->pdev->dev, | 
 | 			 "sensor <%d> THERMAL ALARM stopped: %d < %d\n", | 
 | 			 sensor->id, temp, sensor->thres_temp); | 
 | 	} | 
 |  | 
 | 	return IRQ_HANDLED; | 
 | } | 
 |  | 
 | static int hisi_thermal_register_sensor(struct platform_device *pdev, | 
 | 					struct hisi_thermal_sensor *sensor) | 
 | { | 
 | 	int ret, i; | 
 | 	const struct thermal_trip *trip; | 
 |  | 
 | 	sensor->tzd = devm_thermal_zone_of_sensor_register(&pdev->dev, | 
 | 							   sensor->id, sensor, | 
 | 							   &hisi_of_thermal_ops); | 
 | 	if (IS_ERR(sensor->tzd)) { | 
 | 		ret = PTR_ERR(sensor->tzd); | 
 | 		sensor->tzd = NULL; | 
 | 		dev_err(&pdev->dev, "failed to register sensor id %d: %d\n", | 
 | 			sensor->id, ret); | 
 | 		return ret; | 
 | 	} | 
 |  | 
 | 	trip = of_thermal_get_trip_points(sensor->tzd); | 
 |  | 
 | 	for (i = 0; i < of_thermal_get_ntrips(sensor->tzd); i++) { | 
 | 		if (trip[i].type == THERMAL_TRIP_PASSIVE) { | 
 | 			sensor->thres_temp = trip[i].temperature; | 
 | 			break; | 
 | 		} | 
 | 	} | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static const struct hisi_thermal_ops hi6220_ops = { | 
 | 	.get_temp	= hi6220_thermal_get_temp, | 
 | 	.enable_sensor	= hi6220_thermal_enable_sensor, | 
 | 	.disable_sensor	= hi6220_thermal_disable_sensor, | 
 | 	.irq_handler	= hi6220_thermal_irq_handler, | 
 | 	.probe		= hi6220_thermal_probe, | 
 | }; | 
 |  | 
 | static const struct hisi_thermal_ops hi3660_ops = { | 
 | 	.get_temp	= hi3660_thermal_get_temp, | 
 | 	.enable_sensor	= hi3660_thermal_enable_sensor, | 
 | 	.disable_sensor	= hi3660_thermal_disable_sensor, | 
 | 	.irq_handler	= hi3660_thermal_irq_handler, | 
 | 	.probe		= hi3660_thermal_probe, | 
 | }; | 
 |  | 
 | static const struct of_device_id of_hisi_thermal_match[] = { | 
 | 	{ | 
 | 		.compatible = "hisilicon,tsensor", | 
 | 		.data = &hi6220_ops, | 
 | 	}, | 
 | 	{ | 
 | 		.compatible = "hisilicon,hi3660-tsensor", | 
 | 		.data = &hi3660_ops, | 
 | 	}, | 
 | 	{ /* end */ } | 
 | }; | 
 | MODULE_DEVICE_TABLE(of, of_hisi_thermal_match); | 
 |  | 
 | static void hisi_thermal_toggle_sensor(struct hisi_thermal_sensor *sensor, | 
 | 				       bool on) | 
 | { | 
 | 	struct thermal_zone_device *tzd = sensor->tzd; | 
 |  | 
 | 	if (on) | 
 | 		thermal_zone_device_enable(tzd); | 
 | 	else | 
 | 		thermal_zone_device_disable(tzd); | 
 | } | 
 |  | 
 | static int hisi_thermal_probe(struct platform_device *pdev) | 
 | { | 
 | 	struct hisi_thermal_data *data; | 
 | 	struct device *dev = &pdev->dev; | 
 | 	struct resource *res; | 
 | 	int i, ret; | 
 |  | 
 | 	data = devm_kzalloc(dev, sizeof(*data), GFP_KERNEL); | 
 | 	if (!data) | 
 | 		return -ENOMEM; | 
 |  | 
 | 	data->pdev = pdev; | 
 | 	platform_set_drvdata(pdev, data); | 
 | 	data->ops = of_device_get_match_data(dev); | 
 |  | 
 | 	res = platform_get_resource(pdev, IORESOURCE_MEM, 0); | 
 | 	data->regs = devm_ioremap_resource(dev, res); | 
 | 	if (IS_ERR(data->regs)) | 
 | 		return PTR_ERR(data->regs); | 
 |  | 
 | 	ret = data->ops->probe(data); | 
 | 	if (ret) | 
 | 		return ret; | 
 |  | 
 | 	for (i = 0; i < data->nr_sensors; i++) { | 
 | 		struct hisi_thermal_sensor *sensor = &data->sensor[i]; | 
 |  | 
 | 		ret = hisi_thermal_register_sensor(pdev, sensor); | 
 | 		if (ret) { | 
 | 			dev_err(dev, "failed to register thermal sensor: %d\n", | 
 | 				ret); | 
 | 			return ret; | 
 | 		} | 
 |  | 
 | 		ret = platform_get_irq(pdev, 0); | 
 | 		if (ret < 0) | 
 | 			return ret; | 
 |  | 
 | 		ret = devm_request_threaded_irq(dev, ret, NULL, | 
 | 						hisi_thermal_alarm_irq_thread, | 
 | 						IRQF_ONESHOT, sensor->irq_name, | 
 | 						sensor); | 
 | 		if (ret < 0) { | 
 | 			dev_err(dev, "Failed to request alarm irq: %d\n", ret); | 
 | 			return ret; | 
 | 		} | 
 |  | 
 | 		ret = data->ops->enable_sensor(sensor); | 
 | 		if (ret) { | 
 | 			dev_err(dev, "Failed to setup the sensor: %d\n", ret); | 
 | 			return ret; | 
 | 		} | 
 |  | 
 | 		hisi_thermal_toggle_sensor(sensor, true); | 
 | 	} | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hisi_thermal_remove(struct platform_device *pdev) | 
 | { | 
 | 	struct hisi_thermal_data *data = platform_get_drvdata(pdev); | 
 | 	int i; | 
 |  | 
 | 	for (i = 0; i < data->nr_sensors; i++) { | 
 | 		struct hisi_thermal_sensor *sensor = &data->sensor[i]; | 
 |  | 
 | 		hisi_thermal_toggle_sensor(sensor, false); | 
 | 		data->ops->disable_sensor(sensor); | 
 | 	} | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | #ifdef CONFIG_PM_SLEEP | 
 | static int hisi_thermal_suspend(struct device *dev) | 
 | { | 
 | 	struct hisi_thermal_data *data = dev_get_drvdata(dev); | 
 | 	int i; | 
 |  | 
 | 	for (i = 0; i < data->nr_sensors; i++) | 
 | 		data->ops->disable_sensor(&data->sensor[i]); | 
 |  | 
 | 	return 0; | 
 | } | 
 |  | 
 | static int hisi_thermal_resume(struct device *dev) | 
 | { | 
 | 	struct hisi_thermal_data *data = dev_get_drvdata(dev); | 
 | 	int i, ret = 0; | 
 |  | 
 | 	for (i = 0; i < data->nr_sensors; i++) | 
 | 		ret |= data->ops->enable_sensor(&data->sensor[i]); | 
 |  | 
 | 	return ret; | 
 | } | 
 | #endif | 
 |  | 
 | static SIMPLE_DEV_PM_OPS(hisi_thermal_pm_ops, | 
 | 			 hisi_thermal_suspend, hisi_thermal_resume); | 
 |  | 
 | static struct platform_driver hisi_thermal_driver = { | 
 | 	.driver = { | 
 | 		.name		= "hisi_thermal", | 
 | 		.pm		= &hisi_thermal_pm_ops, | 
 | 		.of_match_table = of_hisi_thermal_match, | 
 | 	}, | 
 | 	.probe	= hisi_thermal_probe, | 
 | 	.remove	= hisi_thermal_remove, | 
 | }; | 
 |  | 
 | module_platform_driver(hisi_thermal_driver); | 
 |  | 
 | MODULE_AUTHOR("Xinwei Kong <kong.kongxinwei@hisilicon.com>"); | 
 | MODULE_AUTHOR("Leo Yan <leo.yan@linaro.org>"); | 
 | MODULE_DESCRIPTION("HiSilicon thermal driver"); | 
 | MODULE_LICENSE("GPL v2"); |