Merge tag 'trace-v4.21-1' of git://git.kernel.org/pub/scm/linux/kernel/git/rostedt...
[sfrench/cifs-2.6.git] / drivers / thermal / armada_thermal.c
1 /*
2  * Marvell EBU Armada SoCs thermal sensor driver
3  *
4  * Copyright (C) 2013 Marvell
5  *
6  * This software is licensed under the terms of the GNU General Public
7  * License version 2, as published by the Free Software Foundation, and
8  * may be copied, distributed, and modified under those terms.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13  * GNU General Public License for more details.
14  *
15  */
16 #include <linux/device.h>
17 #include <linux/err.h>
18 #include <linux/io.h>
19 #include <linux/kernel.h>
20 #include <linux/of.h>
21 #include <linux/module.h>
22 #include <linux/delay.h>
23 #include <linux/platform_device.h>
24 #include <linux/of_device.h>
25 #include <linux/thermal.h>
26 #include <linux/iopoll.h>
27 #include <linux/mfd/syscon.h>
28 #include <linux/regmap.h>
29
30 /* Thermal Manager Control and Status Register */
31 #define PMU_TDC0_SW_RST_MASK            (0x1 << 1)
32 #define PMU_TM_DISABLE_OFFS             0
33 #define PMU_TM_DISABLE_MASK             (0x1 << PMU_TM_DISABLE_OFFS)
34 #define PMU_TDC0_REF_CAL_CNT_OFFS       11
35 #define PMU_TDC0_REF_CAL_CNT_MASK       (0x1ff << PMU_TDC0_REF_CAL_CNT_OFFS)
36 #define PMU_TDC0_OTF_CAL_MASK           (0x1 << 30)
37 #define PMU_TDC0_START_CAL_MASK         (0x1 << 25)
38
39 #define A375_UNIT_CONTROL_SHIFT         27
40 #define A375_UNIT_CONTROL_MASK          0x7
41 #define A375_READOUT_INVERT             BIT(15)
42 #define A375_HW_RESETn                  BIT(8)
43
44 /* Errata fields */
45 #define CONTROL0_TSEN_TC_TRIM_MASK      0x7
46 #define CONTROL0_TSEN_TC_TRIM_VAL       0x3
47
48 #define CONTROL0_TSEN_START             BIT(0)
49 #define CONTROL0_TSEN_RESET             BIT(1)
50 #define CONTROL0_TSEN_ENABLE            BIT(2)
51 #define CONTROL0_TSEN_AVG_BYPASS        BIT(6)
52 #define CONTROL0_TSEN_CHAN_SHIFT        13
53 #define CONTROL0_TSEN_CHAN_MASK         0xF
54 #define CONTROL0_TSEN_OSR_SHIFT         24
55 #define CONTROL0_TSEN_OSR_MAX           0x3
56 #define CONTROL0_TSEN_MODE_SHIFT        30
57 #define CONTROL0_TSEN_MODE_EXTERNAL     0x2
58 #define CONTROL0_TSEN_MODE_MASK         0x3
59
60 #define CONTROL1_TSEN_AVG_SHIFT         0
61 #define CONTROL1_TSEN_AVG_MASK          0x7
62 #define CONTROL1_EXT_TSEN_SW_RESET      BIT(7)
63 #define CONTROL1_EXT_TSEN_HW_RESETn     BIT(8)
64
65 #define STATUS_POLL_PERIOD_US           1000
66 #define STATUS_POLL_TIMEOUT_US          100000
67
68 struct armada_thermal_data;
69
70 /* Marvell EBU Thermal Sensor Dev Structure */
71 struct armada_thermal_priv {
72         struct device *dev;
73         struct regmap *syscon;
74         char zone_name[THERMAL_NAME_LENGTH];
75         /* serialize temperature reads/updates */
76         struct mutex update_lock;
77         struct armada_thermal_data *data;
78         int current_channel;
79 };
80
81 struct armada_thermal_data {
82         /* Initialize the thermal IC */
83         void (*init)(struct platform_device *pdev,
84                      struct armada_thermal_priv *priv);
85
86         /* Formula coeficients: temp = (b - m * reg) / div */
87         s64 coef_b;
88         s64 coef_m;
89         u32 coef_div;
90         bool inverted;
91         bool signed_sample;
92
93         /* Register shift and mask to access the sensor temperature */
94         unsigned int temp_shift;
95         unsigned int temp_mask;
96         u32 is_valid_bit;
97
98         /* Syscon access */
99         unsigned int syscon_control0_off;
100         unsigned int syscon_control1_off;
101         unsigned int syscon_status_off;
102
103         /* One sensor is in the thermal IC, the others are in the CPUs if any */
104         unsigned int cpu_nr;
105 };
106
107 struct armada_drvdata {
108         enum drvtype {
109                 LEGACY,
110                 SYSCON
111         } type;
112         union {
113                 struct armada_thermal_priv *priv;
114                 struct thermal_zone_device *tz;
115         } data;
116 };
117
118 /*
119  * struct armada_thermal_sensor - hold the information of one thermal sensor
120  * @thermal: pointer to the local private structure
121  * @tzd: pointer to the thermal zone device
122  * @id: identifier of the thermal sensor
123  */
124 struct armada_thermal_sensor {
125         struct armada_thermal_priv *priv;
126         int id;
127 };
128
129 static void armadaxp_init(struct platform_device *pdev,
130                           struct armada_thermal_priv *priv)
131 {
132         struct armada_thermal_data *data = priv->data;
133         u32 reg;
134
135         regmap_read(priv->syscon, data->syscon_control1_off, &reg);
136         reg |= PMU_TDC0_OTF_CAL_MASK;
137
138         /* Reference calibration value */
139         reg &= ~PMU_TDC0_REF_CAL_CNT_MASK;
140         reg |= (0xf1 << PMU_TDC0_REF_CAL_CNT_OFFS);
141
142         /* Reset the sensor */
143         reg |= PMU_TDC0_SW_RST_MASK;
144
145         regmap_write(priv->syscon, data->syscon_control1_off, reg);
146
147         /* Enable the sensor */
148         regmap_read(priv->syscon, data->syscon_status_off, &reg);
149         reg &= ~PMU_TM_DISABLE_MASK;
150         regmap_write(priv->syscon, data->syscon_status_off, reg);
151 }
152
153 static void armada370_init(struct platform_device *pdev,
154                            struct armada_thermal_priv *priv)
155 {
156         struct armada_thermal_data *data = priv->data;
157         u32 reg;
158
159         regmap_read(priv->syscon, data->syscon_control1_off, &reg);
160         reg |= PMU_TDC0_OTF_CAL_MASK;
161
162         /* Reference calibration value */
163         reg &= ~PMU_TDC0_REF_CAL_CNT_MASK;
164         reg |= (0xf1 << PMU_TDC0_REF_CAL_CNT_OFFS);
165
166         /* Reset the sensor */
167         reg &= ~PMU_TDC0_START_CAL_MASK;
168
169         regmap_write(priv->syscon, data->syscon_control1_off, reg);
170
171         msleep(10);
172 }
173
174 static void armada375_init(struct platform_device *pdev,
175                            struct armada_thermal_priv *priv)
176 {
177         struct armada_thermal_data *data = priv->data;
178         u32 reg;
179
180         regmap_read(priv->syscon, data->syscon_control1_off, &reg);
181         reg &= ~(A375_UNIT_CONTROL_MASK << A375_UNIT_CONTROL_SHIFT);
182         reg &= ~A375_READOUT_INVERT;
183         reg &= ~A375_HW_RESETn;
184         regmap_write(priv->syscon, data->syscon_control1_off, reg);
185
186         msleep(20);
187
188         reg |= A375_HW_RESETn;
189         regmap_write(priv->syscon, data->syscon_control1_off, reg);
190
191         msleep(50);
192 }
193
194 static int armada_wait_sensor_validity(struct armada_thermal_priv *priv)
195 {
196         u32 reg;
197
198         return regmap_read_poll_timeout(priv->syscon,
199                                         priv->data->syscon_status_off, reg,
200                                         reg & priv->data->is_valid_bit,
201                                         STATUS_POLL_PERIOD_US,
202                                         STATUS_POLL_TIMEOUT_US);
203 }
204
205 static void armada380_init(struct platform_device *pdev,
206                            struct armada_thermal_priv *priv)
207 {
208         struct armada_thermal_data *data = priv->data;
209         u32 reg;
210
211         /* Disable the HW/SW reset */
212         regmap_read(priv->syscon, data->syscon_control1_off, &reg);
213         reg |= CONTROL1_EXT_TSEN_HW_RESETn;
214         reg &= ~CONTROL1_EXT_TSEN_SW_RESET;
215         regmap_write(priv->syscon, data->syscon_control1_off, reg);
216
217         /* Set Tsen Tc Trim to correct default value (errata #132698) */
218         regmap_read(priv->syscon, data->syscon_control0_off, &reg);
219         reg &= ~CONTROL0_TSEN_TC_TRIM_MASK;
220         reg |= CONTROL0_TSEN_TC_TRIM_VAL;
221         regmap_write(priv->syscon, data->syscon_control0_off, reg);
222 }
223
224 static void armada_ap806_init(struct platform_device *pdev,
225                               struct armada_thermal_priv *priv)
226 {
227         struct armada_thermal_data *data = priv->data;
228         u32 reg;
229
230         regmap_read(priv->syscon, data->syscon_control0_off, &reg);
231         reg &= ~CONTROL0_TSEN_RESET;
232         reg |= CONTROL0_TSEN_START | CONTROL0_TSEN_ENABLE;
233
234         /* Sample every ~2ms */
235         reg |= CONTROL0_TSEN_OSR_MAX << CONTROL0_TSEN_OSR_SHIFT;
236
237         /* Enable average (2 samples by default) */
238         reg &= ~CONTROL0_TSEN_AVG_BYPASS;
239
240         regmap_write(priv->syscon, data->syscon_control0_off, reg);
241 }
242
243 static void armada_cp110_init(struct platform_device *pdev,
244                               struct armada_thermal_priv *priv)
245 {
246         struct armada_thermal_data *data = priv->data;
247         u32 reg;
248
249         armada380_init(pdev, priv);
250
251         /* Sample every ~2ms */
252         regmap_read(priv->syscon, data->syscon_control0_off, &reg);
253         reg |= CONTROL0_TSEN_OSR_MAX << CONTROL0_TSEN_OSR_SHIFT;
254         regmap_write(priv->syscon, data->syscon_control0_off, reg);
255
256         /* Average the output value over 2^1 = 2 samples */
257         regmap_read(priv->syscon, data->syscon_control1_off, &reg);
258         reg &= ~CONTROL1_TSEN_AVG_MASK << CONTROL1_TSEN_AVG_SHIFT;
259         reg |= 1 << CONTROL1_TSEN_AVG_SHIFT;
260         regmap_write(priv->syscon, data->syscon_control1_off, reg);
261 }
262
263 static bool armada_is_valid(struct armada_thermal_priv *priv)
264 {
265         u32 reg;
266
267         if (!priv->data->is_valid_bit)
268                 return true;
269
270         regmap_read(priv->syscon, priv->data->syscon_status_off, &reg);
271
272         return reg & priv->data->is_valid_bit;
273 }
274
275 /* There is currently no board with more than one sensor per channel */
276 static int armada_select_channel(struct armada_thermal_priv *priv, int channel)
277 {
278         struct armada_thermal_data *data = priv->data;
279         u32 ctrl0;
280
281         if (channel < 0 || channel > priv->data->cpu_nr)
282                 return -EINVAL;
283
284         if (priv->current_channel == channel)
285                 return 0;
286
287         /* Stop the measurements */
288         regmap_read(priv->syscon, data->syscon_control0_off, &ctrl0);
289         ctrl0 &= ~CONTROL0_TSEN_START;
290         regmap_write(priv->syscon, data->syscon_control0_off, ctrl0);
291
292         /* Reset the mode, internal sensor will be automatically selected */
293         ctrl0 &= ~(CONTROL0_TSEN_MODE_MASK << CONTROL0_TSEN_MODE_SHIFT);
294
295         /* Other channels are external and should be selected accordingly */
296         if (channel) {
297                 /* Change the mode to external */
298                 ctrl0 |= CONTROL0_TSEN_MODE_EXTERNAL <<
299                          CONTROL0_TSEN_MODE_SHIFT;
300                 /* Select the sensor */
301                 ctrl0 &= ~(CONTROL0_TSEN_CHAN_MASK << CONTROL0_TSEN_CHAN_SHIFT);
302                 ctrl0 |= (channel - 1) << CONTROL0_TSEN_CHAN_SHIFT;
303         }
304
305         /* Actually set the mode/channel */
306         regmap_write(priv->syscon, data->syscon_control0_off, ctrl0);
307         priv->current_channel = channel;
308
309         /* Re-start the measurements */
310         ctrl0 |= CONTROL0_TSEN_START;
311         regmap_write(priv->syscon, data->syscon_control0_off, ctrl0);
312
313         /*
314          * The IP has a latency of ~15ms, so after updating the selected source,
315          * we must absolutely wait for the sensor validity bit to ensure we read
316          * actual data.
317          */
318         if (armada_wait_sensor_validity(priv)) {
319                 dev_err(priv->dev,
320                         "Temperature sensor reading not valid\n");
321                 return -EIO;
322         }
323
324         return 0;
325 }
326
327 static int armada_read_sensor(struct armada_thermal_priv *priv, int *temp)
328 {
329         u32 reg, div;
330         s64 sample, b, m;
331
332         regmap_read(priv->syscon, priv->data->syscon_status_off, &reg);
333         reg = (reg >> priv->data->temp_shift) & priv->data->temp_mask;
334         if (priv->data->signed_sample)
335                 /* The most significant bit is the sign bit */
336                 sample = sign_extend32(reg, fls(priv->data->temp_mask) - 1);
337         else
338                 sample = reg;
339
340         /* Get formula coeficients */
341         b = priv->data->coef_b;
342         m = priv->data->coef_m;
343         div = priv->data->coef_div;
344
345         if (priv->data->inverted)
346                 *temp = div_s64((m * sample) - b, div);
347         else
348                 *temp = div_s64(b - (m * sample), div);
349
350         return 0;
351 }
352
353 static int armada_get_temp_legacy(struct thermal_zone_device *thermal,
354                                   int *temp)
355 {
356         struct armada_thermal_priv *priv = thermal->devdata;
357         int ret;
358
359         /* Valid check */
360         if (!armada_is_valid(priv)) {
361                 dev_err(priv->dev,
362                         "Temperature sensor reading not valid\n");
363                 return -EIO;
364         }
365
366         /* Do the actual reading */
367         ret = armada_read_sensor(priv, temp);
368
369         return ret;
370 }
371
372 static struct thermal_zone_device_ops legacy_ops = {
373         .get_temp = armada_get_temp_legacy,
374 };
375
376 static int armada_get_temp(void *_sensor, int *temp)
377 {
378         struct armada_thermal_sensor *sensor = _sensor;
379         struct armada_thermal_priv *priv = sensor->priv;
380         int ret;
381
382         mutex_lock(&priv->update_lock);
383
384         /* Select the desired channel */
385         ret = armada_select_channel(priv, sensor->id);
386         if (ret)
387                 goto unlock_mutex;
388
389         /* Do the actual reading */
390         ret = armada_read_sensor(priv, temp);
391
392 unlock_mutex:
393         mutex_unlock(&priv->update_lock);
394
395         return ret;
396 }
397
398 static const struct thermal_zone_of_device_ops of_ops = {
399         .get_temp = armada_get_temp,
400 };
401
402 static const struct armada_thermal_data armadaxp_data = {
403         .init = armadaxp_init,
404         .temp_shift = 10,
405         .temp_mask = 0x1ff,
406         .coef_b = 3153000000ULL,
407         .coef_m = 10000000ULL,
408         .coef_div = 13825,
409         .syscon_status_off = 0xb0,
410         .syscon_control1_off = 0xd0,
411 };
412
413 static const struct armada_thermal_data armada370_data = {
414         .init = armada370_init,
415         .is_valid_bit = BIT(9),
416         .temp_shift = 10,
417         .temp_mask = 0x1ff,
418         .coef_b = 3153000000ULL,
419         .coef_m = 10000000ULL,
420         .coef_div = 13825,
421         .syscon_status_off = 0x0,
422         .syscon_control1_off = 0x4,
423 };
424
425 static const struct armada_thermal_data armada375_data = {
426         .init = armada375_init,
427         .is_valid_bit = BIT(10),
428         .temp_shift = 0,
429         .temp_mask = 0x1ff,
430         .coef_b = 3171900000ULL,
431         .coef_m = 10000000ULL,
432         .coef_div = 13616,
433         .syscon_status_off = 0x78,
434         .syscon_control0_off = 0x7c,
435         .syscon_control1_off = 0x80,
436 };
437
438 static const struct armada_thermal_data armada380_data = {
439         .init = armada380_init,
440         .is_valid_bit = BIT(10),
441         .temp_shift = 0,
442         .temp_mask = 0x3ff,
443         .coef_b = 1172499100ULL,
444         .coef_m = 2000096ULL,
445         .coef_div = 4201,
446         .inverted = true,
447         .syscon_control0_off = 0x70,
448         .syscon_control1_off = 0x74,
449         .syscon_status_off = 0x78,
450 };
451
452 static const struct armada_thermal_data armada_ap806_data = {
453         .init = armada_ap806_init,
454         .is_valid_bit = BIT(16),
455         .temp_shift = 0,
456         .temp_mask = 0x3ff,
457         .coef_b = -150000LL,
458         .coef_m = 423ULL,
459         .coef_div = 1,
460         .inverted = true,
461         .signed_sample = true,
462         .syscon_control0_off = 0x84,
463         .syscon_control1_off = 0x88,
464         .syscon_status_off = 0x8C,
465         .cpu_nr = 4,
466 };
467
468 static const struct armada_thermal_data armada_cp110_data = {
469         .init = armada_cp110_init,
470         .is_valid_bit = BIT(10),
471         .temp_shift = 0,
472         .temp_mask = 0x3ff,
473         .coef_b = 1172499100ULL,
474         .coef_m = 2000096ULL,
475         .coef_div = 4201,
476         .inverted = true,
477         .syscon_control0_off = 0x70,
478         .syscon_control1_off = 0x74,
479         .syscon_status_off = 0x78,
480 };
481
482 static const struct of_device_id armada_thermal_id_table[] = {
483         {
484                 .compatible = "marvell,armadaxp-thermal",
485                 .data       = &armadaxp_data,
486         },
487         {
488                 .compatible = "marvell,armada370-thermal",
489                 .data       = &armada370_data,
490         },
491         {
492                 .compatible = "marvell,armada375-thermal",
493                 .data       = &armada375_data,
494         },
495         {
496                 .compatible = "marvell,armada380-thermal",
497                 .data       = &armada380_data,
498         },
499         {
500                 .compatible = "marvell,armada-ap806-thermal",
501                 .data       = &armada_ap806_data,
502         },
503         {
504                 .compatible = "marvell,armada-cp110-thermal",
505                 .data       = &armada_cp110_data,
506         },
507         {
508                 /* sentinel */
509         },
510 };
511 MODULE_DEVICE_TABLE(of, armada_thermal_id_table);
512
513 static const struct regmap_config armada_thermal_regmap_config = {
514         .reg_bits = 32,
515         .reg_stride = 4,
516         .val_bits = 32,
517         .fast_io = true,
518 };
519
520 static int armada_thermal_probe_legacy(struct platform_device *pdev,
521                                        struct armada_thermal_priv *priv)
522 {
523         struct armada_thermal_data *data = priv->data;
524         struct resource *res;
525         void __iomem *base;
526
527         /* First memory region points towards the status register */
528         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
529         base = devm_ioremap_resource(&pdev->dev, res);
530         if (IS_ERR(base))
531                 return PTR_ERR(base);
532
533         /*
534          * Fix up from the old individual DT register specification to
535          * cover all the registers.  We do this by adjusting the ioremap()
536          * result, which should be fine as ioremap() deals with pages.
537          * However, validate that we do not cross a page boundary while
538          * making this adjustment.
539          */
540         if (((unsigned long)base & ~PAGE_MASK) < data->syscon_status_off)
541                 return -EINVAL;
542         base -= data->syscon_status_off;
543
544         priv->syscon = devm_regmap_init_mmio(&pdev->dev, base,
545                                              &armada_thermal_regmap_config);
546         if (IS_ERR(priv->syscon))
547                 return PTR_ERR(priv->syscon);
548
549         return 0;
550 }
551
552 static int armada_thermal_probe_syscon(struct platform_device *pdev,
553                                        struct armada_thermal_priv *priv)
554 {
555         priv->syscon = syscon_node_to_regmap(pdev->dev.parent->of_node);
556         if (IS_ERR(priv->syscon))
557                 return PTR_ERR(priv->syscon);
558
559         return 0;
560 }
561
562 static void armada_set_sane_name(struct platform_device *pdev,
563                                  struct armada_thermal_priv *priv)
564 {
565         const char *name = dev_name(&pdev->dev);
566         char *insane_char;
567
568         if (strlen(name) > THERMAL_NAME_LENGTH) {
569                 /*
570                  * When inside a system controller, the device name has the
571                  * form: f06f8000.system-controller:ap-thermal so stripping
572                  * after the ':' should give us a shorter but meaningful name.
573                  */
574                 name = strrchr(name, ':');
575                 if (!name)
576                         name = "armada_thermal";
577                 else
578                         name++;
579         }
580
581         /* Save the name locally */
582         strncpy(priv->zone_name, name, THERMAL_NAME_LENGTH - 1);
583         priv->zone_name[THERMAL_NAME_LENGTH - 1] = '\0';
584
585         /* Then check there are no '-' or hwmon core will complain */
586         do {
587                 insane_char = strpbrk(priv->zone_name, "-");
588                 if (insane_char)
589                         *insane_char = '_';
590         } while (insane_char);
591 }
592
593 static int armada_thermal_probe(struct platform_device *pdev)
594 {
595         struct thermal_zone_device *tz;
596         struct armada_thermal_sensor *sensor;
597         struct armada_drvdata *drvdata;
598         const struct of_device_id *match;
599         struct armada_thermal_priv *priv;
600         int sensor_id;
601         int ret;
602
603         match = of_match_device(armada_thermal_id_table, &pdev->dev);
604         if (!match)
605                 return -ENODEV;
606
607         priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
608         if (!priv)
609                 return -ENOMEM;
610
611         drvdata = devm_kzalloc(&pdev->dev, sizeof(*drvdata), GFP_KERNEL);
612         if (!drvdata)
613                 return -ENOMEM;
614
615         priv->dev = &pdev->dev;
616         priv->data = (struct armada_thermal_data *)match->data;
617
618         mutex_init(&priv->update_lock);
619
620         /*
621          * Legacy DT bindings only described "control1" register (also referred
622          * as "control MSB" on old documentation). Then, bindings moved to cover
623          * "control0/control LSB" and "control1/control MSB" registers within
624          * the same resource, which was then of size 8 instead of 4.
625          *
626          * The logic of defining sporadic registers is broken. For instance, it
627          * blocked the addition of the overheat interrupt feature that needed
628          * another resource somewhere else in the same memory area. One solution
629          * is to define an overall system controller and put the thermal node
630          * into it, which requires the use of regmaps across all the driver.
631          */
632         if (IS_ERR(syscon_node_to_regmap(pdev->dev.parent->of_node))) {
633                 /* Ensure device name is correct for the thermal core */
634                 armada_set_sane_name(pdev, priv);
635
636                 ret = armada_thermal_probe_legacy(pdev, priv);
637                 if (ret)
638                         return ret;
639
640                 priv->data->init(pdev, priv);
641
642                 /* Wait the sensors to be valid */
643                 armada_wait_sensor_validity(priv);
644
645                 tz = thermal_zone_device_register(priv->zone_name, 0, 0, priv,
646                                                   &legacy_ops, NULL, 0, 0);
647                 if (IS_ERR(tz)) {
648                         dev_err(&pdev->dev,
649                                 "Failed to register thermal zone device\n");
650                         return PTR_ERR(tz);
651                 }
652
653                 drvdata->type = LEGACY;
654                 drvdata->data.tz = tz;
655                 platform_set_drvdata(pdev, drvdata);
656
657                 return 0;
658         }
659
660         ret = armada_thermal_probe_syscon(pdev, priv);
661         if (ret)
662                 return ret;
663
664         priv->current_channel = -1;
665         priv->data->init(pdev, priv);
666         drvdata->type = SYSCON;
667         drvdata->data.priv = priv;
668         platform_set_drvdata(pdev, drvdata);
669
670         /*
671          * There is one channel for the IC and one per CPU (if any), each
672          * channel has one sensor.
673          */
674         for (sensor_id = 0; sensor_id <= priv->data->cpu_nr; sensor_id++) {
675                 sensor = devm_kzalloc(&pdev->dev,
676                                       sizeof(struct armada_thermal_sensor),
677                                       GFP_KERNEL);
678                 if (!sensor)
679                         return -ENOMEM;
680
681                 /* Register the sensor */
682                 sensor->priv = priv;
683                 sensor->id = sensor_id;
684                 tz = devm_thermal_zone_of_sensor_register(&pdev->dev,
685                                                           sensor->id, sensor,
686                                                           &of_ops);
687                 if (IS_ERR(tz)) {
688                         dev_info(&pdev->dev, "Thermal sensor %d unavailable\n",
689                                  sensor_id);
690                         devm_kfree(&pdev->dev, sensor);
691                         continue;
692                 }
693         }
694
695         return 0;
696 }
697
698 static int armada_thermal_exit(struct platform_device *pdev)
699 {
700         struct armada_drvdata *drvdata = platform_get_drvdata(pdev);
701
702         if (drvdata->type == LEGACY)
703                 thermal_zone_device_unregister(drvdata->data.tz);
704
705         return 0;
706 }
707
708 static struct platform_driver armada_thermal_driver = {
709         .probe = armada_thermal_probe,
710         .remove = armada_thermal_exit,
711         .driver = {
712                 .name = "armada_thermal",
713                 .of_match_table = armada_thermal_id_table,
714         },
715 };
716
717 module_platform_driver(armada_thermal_driver);
718
719 MODULE_AUTHOR("Ezequiel Garcia <ezequiel.garcia@free-electrons.com>");
720 MODULE_DESCRIPTION("Marvell EBU Armada SoCs thermal driver");
721 MODULE_LICENSE("GPL v2");