xref: /openbmc/linux/drivers/hte/hte-tegra194.c (revision 2dd6532e)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2021-2022 NVIDIA Corporation
4  *
5  * Author: Dipen Patel <dipenp@nvidia.com>
6  */
7 
8 #include <linux/err.h>
9 #include <linux/io.h>
10 #include <linux/module.h>
11 #include <linux/slab.h>
12 #include <linux/stat.h>
13 #include <linux/interrupt.h>
14 #include <linux/of.h>
15 #include <linux/of_device.h>
16 #include <linux/platform_device.h>
17 #include <linux/hte.h>
18 #include <linux/uaccess.h>
19 #include <linux/gpio/driver.h>
20 #include <linux/gpio/consumer.h>
21 
22 #define HTE_SUSPEND	0
23 
24 /* HTE source clock TSC is 31.25MHz */
25 #define HTE_TS_CLK_RATE_HZ	31250000ULL
26 #define HTE_CLK_RATE_NS		32
27 #define HTE_TS_NS_SHIFT	__builtin_ctz(HTE_CLK_RATE_NS)
28 
29 #define NV_AON_SLICE_INVALID	-1
30 #define NV_LINES_IN_SLICE	32
31 
32 /* AON HTE line map For slice 1 */
33 #define NV_AON_HTE_SLICE1_IRQ_GPIO_28	12
34 #define NV_AON_HTE_SLICE1_IRQ_GPIO_29	13
35 
36 /* AON HTE line map For slice 2 */
37 #define NV_AON_HTE_SLICE2_IRQ_GPIO_0	0
38 #define NV_AON_HTE_SLICE2_IRQ_GPIO_1	1
39 #define NV_AON_HTE_SLICE2_IRQ_GPIO_2	2
40 #define NV_AON_HTE_SLICE2_IRQ_GPIO_3	3
41 #define NV_AON_HTE_SLICE2_IRQ_GPIO_4	4
42 #define NV_AON_HTE_SLICE2_IRQ_GPIO_5	5
43 #define NV_AON_HTE_SLICE2_IRQ_GPIO_6	6
44 #define NV_AON_HTE_SLICE2_IRQ_GPIO_7	7
45 #define NV_AON_HTE_SLICE2_IRQ_GPIO_8	8
46 #define NV_AON_HTE_SLICE2_IRQ_GPIO_9	9
47 #define NV_AON_HTE_SLICE2_IRQ_GPIO_10	10
48 #define NV_AON_HTE_SLICE2_IRQ_GPIO_11	11
49 #define NV_AON_HTE_SLICE2_IRQ_GPIO_12	12
50 #define NV_AON_HTE_SLICE2_IRQ_GPIO_13	13
51 #define NV_AON_HTE_SLICE2_IRQ_GPIO_14	14
52 #define NV_AON_HTE_SLICE2_IRQ_GPIO_15	15
53 #define NV_AON_HTE_SLICE2_IRQ_GPIO_16	16
54 #define NV_AON_HTE_SLICE2_IRQ_GPIO_17	17
55 #define NV_AON_HTE_SLICE2_IRQ_GPIO_18	18
56 #define NV_AON_HTE_SLICE2_IRQ_GPIO_19	19
57 #define NV_AON_HTE_SLICE2_IRQ_GPIO_20	20
58 #define NV_AON_HTE_SLICE2_IRQ_GPIO_21	21
59 #define NV_AON_HTE_SLICE2_IRQ_GPIO_22	22
60 #define NV_AON_HTE_SLICE2_IRQ_GPIO_23	23
61 #define NV_AON_HTE_SLICE2_IRQ_GPIO_24	24
62 #define NV_AON_HTE_SLICE2_IRQ_GPIO_25	25
63 #define NV_AON_HTE_SLICE2_IRQ_GPIO_26	26
64 #define NV_AON_HTE_SLICE2_IRQ_GPIO_27	27
65 
66 #define HTE_TECTRL		0x0
67 #define HTE_TETSCH		0x4
68 #define HTE_TETSCL		0x8
69 #define HTE_TESRC		0xC
70 #define HTE_TECCV		0x10
71 #define HTE_TEPCV		0x14
72 #define HTE_TECMD		0x1C
73 #define HTE_TESTATUS		0x20
74 #define HTE_SLICE0_TETEN	0x40
75 #define HTE_SLICE1_TETEN	0x60
76 
77 #define HTE_SLICE_SIZE		(HTE_SLICE1_TETEN - HTE_SLICE0_TETEN)
78 
79 #define HTE_TECTRL_ENABLE_ENABLE	0x1
80 
81 #define HTE_TECTRL_OCCU_SHIFT		0x8
82 #define HTE_TECTRL_INTR_SHIFT		0x1
83 #define HTE_TECTRL_INTR_ENABLE		0x1
84 
85 #define HTE_TESRC_SLICE_SHIFT		16
86 #define HTE_TESRC_SLICE_DEFAULT_MASK	0xFF
87 
88 #define HTE_TECMD_CMD_POP		0x1
89 
90 #define HTE_TESTATUS_OCCUPANCY_SHIFT	8
91 #define HTE_TESTATUS_OCCUPANCY_MASK	0xFF
92 
93 enum tegra_hte_type {
94 	HTE_TEGRA_TYPE_GPIO = 1U << 0,
95 	HTE_TEGRA_TYPE_LIC = 1U << 1,
96 };
97 
98 struct hte_slices {
99 	u32 r_val;
100 	unsigned long flags;
101 	/* to prevent lines mapped to same slice updating its register */
102 	spinlock_t s_lock;
103 };
104 
105 struct tegra_hte_line_mapped {
106 	int slice;
107 	u32 bit_index;
108 };
109 
110 struct tegra_hte_line_data {
111 	unsigned long flags;
112 	void *data;
113 };
114 
115 struct tegra_hte_data {
116 	enum tegra_hte_type type;
117 	u32 map_sz;
118 	u32 sec_map_sz;
119 	const struct tegra_hte_line_mapped *map;
120 	const struct tegra_hte_line_mapped *sec_map;
121 };
122 
123 struct tegra_hte_soc {
124 	int hte_irq;
125 	u32 itr_thrshld;
126 	u32 conf_rval;
127 	struct hte_slices *sl;
128 	const struct tegra_hte_data *prov_data;
129 	struct tegra_hte_line_data *line_data;
130 	struct hte_chip *chip;
131 	struct gpio_chip *c;
132 	void __iomem *regs;
133 };
134 
135 static const struct tegra_hte_line_mapped tegra194_aon_gpio_map[] = {
136 	/* gpio, slice, bit_index */
137 	/* AA port */
138 	[0]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_11},
139 	[1]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_10},
140 	[2]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_9},
141 	[3]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_8},
142 	[4]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_7},
143 	[5]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_6},
144 	[6]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_5},
145 	[7]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_4},
146 	/* BB port */
147 	[8]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_3},
148 	[9]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_2},
149 	[10] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_1},
150 	[11] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_0},
151 	/* CC port */
152 	[12] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_22},
153 	[13] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_21},
154 	[14] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_20},
155 	[15] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_19},
156 	[16] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_18},
157 	[17] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_17},
158 	[18] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_16},
159 	[19] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_15},
160 	/* DD port */
161 	[20] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_14},
162 	[21] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_13},
163 	[22] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_12},
164 	/* EE port */
165 	[23] = {1, NV_AON_HTE_SLICE1_IRQ_GPIO_29},
166 	[24] = {1, NV_AON_HTE_SLICE1_IRQ_GPIO_28},
167 	[25] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_27},
168 	[26] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_26},
169 	[27] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_25},
170 	[28] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_24},
171 	[29] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_23},
172 };
173 
174 static const struct tegra_hte_line_mapped tegra194_aon_gpio_sec_map[] = {
175 	/* gpio, slice, bit_index */
176 	/* AA port */
177 	[0]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_11},
178 	[1]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_10},
179 	[2]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_9},
180 	[3]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_8},
181 	[4]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_7},
182 	[5]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_6},
183 	[6]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_5},
184 	[7]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_4},
185 	/* BB port */
186 	[8]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_3},
187 	[9]  = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_2},
188 	[10] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_1},
189 	[11] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_0},
190 	[12]  = {NV_AON_SLICE_INVALID, 0},
191 	[13]  = {NV_AON_SLICE_INVALID, 0},
192 	[14] = {NV_AON_SLICE_INVALID, 0},
193 	[15] = {NV_AON_SLICE_INVALID, 0},
194 	/* CC port */
195 	[16] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_22},
196 	[17] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_21},
197 	[18] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_20},
198 	[19] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_19},
199 	[20] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_18},
200 	[21] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_17},
201 	[22] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_16},
202 	[23] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_15},
203 	/* DD port */
204 	[24] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_14},
205 	[25] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_13},
206 	[26] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_12},
207 	[27] = {NV_AON_SLICE_INVALID, 0},
208 	[28] = {NV_AON_SLICE_INVALID, 0},
209 	[29] = {NV_AON_SLICE_INVALID, 0},
210 	[30] = {NV_AON_SLICE_INVALID, 0},
211 	[31] = {NV_AON_SLICE_INVALID, 0},
212 	/* EE port */
213 	[32] = {1, NV_AON_HTE_SLICE1_IRQ_GPIO_29},
214 	[33] = {1, NV_AON_HTE_SLICE1_IRQ_GPIO_28},
215 	[34] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_27},
216 	[35] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_26},
217 	[36] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_25},
218 	[37] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_24},
219 	[38] = {2, NV_AON_HTE_SLICE2_IRQ_GPIO_23},
220 	[39] = {NV_AON_SLICE_INVALID, 0},
221 };
222 
223 static const struct tegra_hte_data aon_hte = {
224 	.map_sz = ARRAY_SIZE(tegra194_aon_gpio_map),
225 	.map = tegra194_aon_gpio_map,
226 	.sec_map_sz = ARRAY_SIZE(tegra194_aon_gpio_sec_map),
227 	.sec_map = tegra194_aon_gpio_sec_map,
228 	.type = HTE_TEGRA_TYPE_GPIO,
229 };
230 
231 static const struct tegra_hte_data lic_hte = {
232 	.map_sz = 0,
233 	.map = NULL,
234 	.type = HTE_TEGRA_TYPE_LIC,
235 };
236 
237 static inline u32 tegra_hte_readl(struct tegra_hte_soc *hte, u32 reg)
238 {
239 	return readl(hte->regs + reg);
240 }
241 
242 static inline void tegra_hte_writel(struct tegra_hte_soc *hte, u32 reg,
243 				    u32 val)
244 {
245 	writel(val, hte->regs + reg);
246 }
247 
248 static int tegra_hte_map_to_line_id(u32 eid,
249 				    const struct tegra_hte_line_mapped *m,
250 				    u32 map_sz, u32 *mapped)
251 {
252 
253 	if (m) {
254 		if (eid > map_sz)
255 			return -EINVAL;
256 		if (m[eid].slice == NV_AON_SLICE_INVALID)
257 			return -EINVAL;
258 
259 		*mapped = (m[eid].slice << 5) + m[eid].bit_index;
260 	} else {
261 		*mapped = eid;
262 	}
263 
264 	return 0;
265 }
266 
267 static int tegra_hte_line_xlate(struct hte_chip *gc,
268 				const struct of_phandle_args *args,
269 				struct hte_ts_desc *desc, u32 *xlated_id)
270 {
271 	int ret = 0;
272 	u32 line_id;
273 	struct tegra_hte_soc *gs;
274 	const struct tegra_hte_line_mapped *map = NULL;
275 	u32 map_sz = 0;
276 
277 	if (!gc || !desc || !xlated_id)
278 		return -EINVAL;
279 
280 	if (args) {
281 		if (gc->of_hte_n_cells < 1)
282 			return -EINVAL;
283 
284 		if (args->args_count != gc->of_hte_n_cells)
285 			return -EINVAL;
286 
287 		desc->attr.line_id = args->args[0];
288 	}
289 
290 	gs = gc->data;
291 	if (!gs || !gs->prov_data)
292 		return -EINVAL;
293 
294 	/*
295 	 *
296 	 * There are two paths GPIO consumers can take as follows:
297 	 * 1) The consumer (gpiolib-cdev for example) which uses GPIO global
298 	 * number which gets assigned run time.
299 	 * 2) The consumer passing GPIO from the DT which is assigned
300 	 * statically for example by using TEGRA194_AON_GPIO gpio DT binding.
301 	 *
302 	 * The code below addresses both the consumer use cases and maps into
303 	 * HTE/GTE namespace.
304 	 */
305 	if (gs->prov_data->type == HTE_TEGRA_TYPE_GPIO && !args) {
306 		line_id = desc->attr.line_id - gs->c->base;
307 		map = gs->prov_data->map;
308 		map_sz = gs->prov_data->map_sz;
309 	} else if (gs->prov_data->type == HTE_TEGRA_TYPE_GPIO && args) {
310 		line_id = desc->attr.line_id;
311 		map = gs->prov_data->sec_map;
312 		map_sz = gs->prov_data->sec_map_sz;
313 	} else {
314 		line_id = desc->attr.line_id;
315 	}
316 
317 	ret = tegra_hte_map_to_line_id(line_id, map, map_sz, xlated_id);
318 	if (ret < 0) {
319 		dev_err(gc->dev, "line_id:%u mapping failed\n",
320 			desc->attr.line_id);
321 		return ret;
322 	}
323 
324 	if (*xlated_id > gc->nlines)
325 		return -EINVAL;
326 
327 	dev_dbg(gc->dev, "requested id:%u, xlated id:%u\n",
328 		desc->attr.line_id, *xlated_id);
329 
330 	return 0;
331 }
332 
333 static int tegra_hte_line_xlate_plat(struct hte_chip *gc,
334 				     struct hte_ts_desc *desc, u32 *xlated_id)
335 {
336 	return tegra_hte_line_xlate(gc, NULL, desc, xlated_id);
337 }
338 
339 static int tegra_hte_en_dis_common(struct hte_chip *chip, u32 line_id, bool en)
340 {
341 	u32 slice, sl_bit_shift, line_bit, val, reg;
342 	struct tegra_hte_soc *gs;
343 
344 	sl_bit_shift = __builtin_ctz(HTE_SLICE_SIZE);
345 
346 	if (!chip)
347 		return -EINVAL;
348 
349 	gs = chip->data;
350 
351 	if (line_id > chip->nlines) {
352 		dev_err(chip->dev,
353 			"line id: %u is not supported by this controller\n",
354 			line_id);
355 		return -EINVAL;
356 	}
357 
358 	slice = line_id >> sl_bit_shift;
359 	line_bit = line_id & (HTE_SLICE_SIZE - 1);
360 	reg = (slice << sl_bit_shift) + HTE_SLICE0_TETEN;
361 
362 	spin_lock(&gs->sl[slice].s_lock);
363 
364 	if (test_bit(HTE_SUSPEND, &gs->sl[slice].flags)) {
365 		spin_unlock(&gs->sl[slice].s_lock);
366 		dev_dbg(chip->dev, "device suspended");
367 		return -EBUSY;
368 	}
369 
370 	val = tegra_hte_readl(gs, reg);
371 	if (en)
372 		val = val | (1 << line_bit);
373 	else
374 		val = val & (~(1 << line_bit));
375 	tegra_hte_writel(gs, reg, val);
376 
377 	spin_unlock(&gs->sl[slice].s_lock);
378 
379 	dev_dbg(chip->dev, "line: %u, slice %u, line_bit %u, reg:0x%x\n",
380 		line_id, slice, line_bit, reg);
381 
382 	return 0;
383 }
384 
385 static int tegra_hte_enable(struct hte_chip *chip, u32 line_id)
386 {
387 	if (!chip)
388 		return -EINVAL;
389 
390 	return tegra_hte_en_dis_common(chip, line_id, true);
391 }
392 
393 static int tegra_hte_disable(struct hte_chip *chip, u32 line_id)
394 {
395 	if (!chip)
396 		return -EINVAL;
397 
398 	return tegra_hte_en_dis_common(chip, line_id, false);
399 }
400 
401 static int tegra_hte_request(struct hte_chip *chip, struct hte_ts_desc *desc,
402 			     u32 line_id)
403 {
404 	int ret;
405 	struct tegra_hte_soc *gs;
406 	struct hte_line_attr *attr;
407 
408 	if (!chip || !chip->data || !desc)
409 		return -EINVAL;
410 
411 	gs = chip->data;
412 	attr = &desc->attr;
413 
414 	if (gs->prov_data->type == HTE_TEGRA_TYPE_GPIO) {
415 		if (!attr->line_data)
416 			return -EINVAL;
417 
418 		ret = gpiod_enable_hw_timestamp_ns(attr->line_data,
419 						   attr->edge_flags);
420 		if (ret)
421 			return ret;
422 
423 		gs->line_data[line_id].data = attr->line_data;
424 		gs->line_data[line_id].flags = attr->edge_flags;
425 	}
426 
427 	return tegra_hte_en_dis_common(chip, line_id, true);
428 }
429 
430 static int tegra_hte_release(struct hte_chip *chip, struct hte_ts_desc *desc,
431 			     u32 line_id)
432 {
433 	struct tegra_hte_soc *gs;
434 	struct hte_line_attr *attr;
435 	int ret;
436 
437 	if (!chip || !chip->data || !desc)
438 		return -EINVAL;
439 
440 	gs = chip->data;
441 	attr = &desc->attr;
442 
443 	if (gs->prov_data->type == HTE_TEGRA_TYPE_GPIO) {
444 		ret = gpiod_disable_hw_timestamp_ns(attr->line_data,
445 						    gs->line_data[line_id].flags);
446 		if (ret)
447 			return ret;
448 
449 		gs->line_data[line_id].data = NULL;
450 		gs->line_data[line_id].flags = 0;
451 	}
452 
453 	return tegra_hte_en_dis_common(chip, line_id, false);
454 }
455 
456 static int tegra_hte_clk_src_info(struct hte_chip *chip,
457 				  struct hte_clk_info *ci)
458 {
459 	(void)chip;
460 
461 	if (!ci)
462 		return -EINVAL;
463 
464 	ci->hz = HTE_TS_CLK_RATE_HZ;
465 	ci->type = CLOCK_MONOTONIC;
466 
467 	return 0;
468 }
469 
470 static int tegra_hte_get_level(struct tegra_hte_soc *gs, u32 line_id)
471 {
472 	struct gpio_desc *desc;
473 
474 	if (gs->prov_data->type == HTE_TEGRA_TYPE_GPIO) {
475 		desc = gs->line_data[line_id].data;
476 		if (desc)
477 			return gpiod_get_raw_value(desc);
478 	}
479 
480 	return -1;
481 }
482 
483 static void tegra_hte_read_fifo(struct tegra_hte_soc *gs)
484 {
485 	u32 tsh, tsl, src, pv, cv, acv, slice, bit_index, line_id;
486 	u64 tsc;
487 	struct hte_ts_data el;
488 
489 	while ((tegra_hte_readl(gs, HTE_TESTATUS) >>
490 		HTE_TESTATUS_OCCUPANCY_SHIFT) &
491 		HTE_TESTATUS_OCCUPANCY_MASK) {
492 		tsh = tegra_hte_readl(gs, HTE_TETSCH);
493 		tsl = tegra_hte_readl(gs, HTE_TETSCL);
494 		tsc = (((u64)tsh << 32) | tsl);
495 
496 		src = tegra_hte_readl(gs, HTE_TESRC);
497 		slice = (src >> HTE_TESRC_SLICE_SHIFT) &
498 			    HTE_TESRC_SLICE_DEFAULT_MASK;
499 
500 		pv = tegra_hte_readl(gs, HTE_TEPCV);
501 		cv = tegra_hte_readl(gs, HTE_TECCV);
502 		acv = pv ^ cv;
503 		while (acv) {
504 			bit_index = __builtin_ctz(acv);
505 			line_id = bit_index + (slice << 5);
506 			el.tsc = tsc << HTE_TS_NS_SHIFT;
507 			el.raw_level = tegra_hte_get_level(gs, line_id);
508 			hte_push_ts_ns(gs->chip, line_id, &el);
509 			acv &= ~BIT(bit_index);
510 		}
511 		tegra_hte_writel(gs, HTE_TECMD, HTE_TECMD_CMD_POP);
512 	}
513 }
514 
515 static irqreturn_t tegra_hte_isr(int irq, void *dev_id)
516 {
517 	struct tegra_hte_soc *gs = dev_id;
518 	(void)irq;
519 
520 	tegra_hte_read_fifo(gs);
521 
522 	return IRQ_HANDLED;
523 }
524 
525 static bool tegra_hte_match_from_linedata(const struct hte_chip *chip,
526 					  const struct hte_ts_desc *hdesc)
527 {
528 	struct tegra_hte_soc *hte_dev = chip->data;
529 
530 	if (!hte_dev || (hte_dev->prov_data->type != HTE_TEGRA_TYPE_GPIO))
531 		return false;
532 
533 	return hte_dev->c == gpiod_to_chip(hdesc->attr.line_data);
534 }
535 
536 static const struct of_device_id tegra_hte_of_match[] = {
537 	{ .compatible = "nvidia,tegra194-gte-lic", .data = &lic_hte},
538 	{ .compatible = "nvidia,tegra194-gte-aon", .data = &aon_hte},
539 	{ }
540 };
541 MODULE_DEVICE_TABLE(of, tegra_hte_of_match);
542 
543 static const struct hte_ops g_ops = {
544 	.request = tegra_hte_request,
545 	.release = tegra_hte_release,
546 	.enable = tegra_hte_enable,
547 	.disable = tegra_hte_disable,
548 	.get_clk_src_info = tegra_hte_clk_src_info,
549 };
550 
551 static void tegra_gte_disable(void *data)
552 {
553 	struct platform_device *pdev = data;
554 	struct tegra_hte_soc *gs = dev_get_drvdata(&pdev->dev);
555 
556 	tegra_hte_writel(gs, HTE_TECTRL, 0);
557 }
558 
559 static int tegra_get_gpiochip_from_name(struct gpio_chip *chip, void *data)
560 {
561 	return !strcmp(chip->label, data);
562 }
563 
564 static int tegra_hte_probe(struct platform_device *pdev)
565 {
566 	int ret;
567 	u32 i, slices, val = 0;
568 	u32 nlines;
569 	struct device *dev;
570 	struct tegra_hte_soc *hte_dev;
571 	struct hte_chip *gc;
572 
573 	dev = &pdev->dev;
574 
575 	ret = of_property_read_u32(dev->of_node, "nvidia,slices", &slices);
576 	if (ret != 0) {
577 		dev_err(dev, "Could not read slices\n");
578 		return -EINVAL;
579 	}
580 	nlines = slices << 5;
581 
582 	hte_dev = devm_kzalloc(dev, sizeof(*hte_dev), GFP_KERNEL);
583 	if (!hte_dev)
584 		return -ENOMEM;
585 
586 	gc = devm_kzalloc(dev, sizeof(*gc), GFP_KERNEL);
587 	if (!gc)
588 		return -ENOMEM;
589 
590 	dev_set_drvdata(&pdev->dev, hte_dev);
591 	hte_dev->prov_data = of_device_get_match_data(&pdev->dev);
592 
593 	hte_dev->regs = devm_platform_ioremap_resource(pdev, 0);
594 	if (IS_ERR(hte_dev->regs))
595 		return PTR_ERR(hte_dev->regs);
596 
597 	ret = of_property_read_u32(dev->of_node, "nvidia,int-threshold",
598 				   &hte_dev->itr_thrshld);
599 	if (ret != 0)
600 		hte_dev->itr_thrshld = 1;
601 
602 	hte_dev->sl = devm_kcalloc(dev, slices, sizeof(*hte_dev->sl),
603 				   GFP_KERNEL);
604 	if (!hte_dev->sl)
605 		return -ENOMEM;
606 
607 	ret = platform_get_irq(pdev, 0);
608 	if (ret < 0) {
609 		dev_err_probe(dev, ret, "failed to get irq\n");
610 		return ret;
611 	}
612 	hte_dev->hte_irq = ret;
613 	ret = devm_request_irq(dev, hte_dev->hte_irq, tegra_hte_isr, 0,
614 			       dev_name(dev), hte_dev);
615 	if (ret < 0) {
616 		dev_err(dev, "request irq failed.\n");
617 		return ret;
618 	}
619 
620 	gc->nlines = nlines;
621 	gc->ops = &g_ops;
622 	gc->dev = dev;
623 	gc->data = hte_dev;
624 	gc->xlate_of = tegra_hte_line_xlate;
625 	gc->xlate_plat = tegra_hte_line_xlate_plat;
626 	gc->of_hte_n_cells = 1;
627 
628 	if (hte_dev->prov_data &&
629 	    hte_dev->prov_data->type == HTE_TEGRA_TYPE_GPIO) {
630 		hte_dev->line_data = devm_kcalloc(dev, nlines,
631 						  sizeof(*hte_dev->line_data),
632 						  GFP_KERNEL);
633 		if (!hte_dev->line_data)
634 			return -ENOMEM;
635 
636 		gc->match_from_linedata = tegra_hte_match_from_linedata;
637 
638 		hte_dev->c = gpiochip_find("tegra194-gpio-aon",
639 					   tegra_get_gpiochip_from_name);
640 		if (!hte_dev->c)
641 			return dev_err_probe(dev, -EPROBE_DEFER,
642 					     "wait for gpio controller\n");
643 	}
644 
645 	hte_dev->chip = gc;
646 
647 	ret = devm_hte_register_chip(hte_dev->chip);
648 	if (ret) {
649 		dev_err(gc->dev, "hte chip register failed");
650 		return ret;
651 	}
652 
653 	for (i = 0; i < slices; i++) {
654 		hte_dev->sl[i].flags = 0;
655 		spin_lock_init(&hte_dev->sl[i].s_lock);
656 	}
657 
658 	val = HTE_TECTRL_ENABLE_ENABLE |
659 	      (HTE_TECTRL_INTR_ENABLE << HTE_TECTRL_INTR_SHIFT) |
660 	      (hte_dev->itr_thrshld << HTE_TECTRL_OCCU_SHIFT);
661 	tegra_hte_writel(hte_dev, HTE_TECTRL, val);
662 
663 	ret = devm_add_action_or_reset(&pdev->dev, tegra_gte_disable, pdev);
664 	if (ret)
665 		return ret;
666 
667 	dev_dbg(gc->dev, "lines: %d, slices:%d", gc->nlines, slices);
668 
669 	return 0;
670 }
671 
672 static int __maybe_unused tegra_hte_resume_early(struct device *dev)
673 {
674 	u32 i;
675 	struct tegra_hte_soc *gs = dev_get_drvdata(dev);
676 	u32 slices = gs->chip->nlines / NV_LINES_IN_SLICE;
677 	u32 sl_bit_shift = __builtin_ctz(HTE_SLICE_SIZE);
678 
679 	tegra_hte_writel(gs, HTE_TECTRL, gs->conf_rval);
680 
681 	for (i = 0; i < slices; i++) {
682 		spin_lock(&gs->sl[i].s_lock);
683 		tegra_hte_writel(gs,
684 				 ((i << sl_bit_shift) + HTE_SLICE0_TETEN),
685 				 gs->sl[i].r_val);
686 		clear_bit(HTE_SUSPEND, &gs->sl[i].flags);
687 		spin_unlock(&gs->sl[i].s_lock);
688 	}
689 
690 	return 0;
691 }
692 
693 static int __maybe_unused tegra_hte_suspend_late(struct device *dev)
694 {
695 	u32 i;
696 	struct tegra_hte_soc *gs = dev_get_drvdata(dev);
697 	u32 slices = gs->chip->nlines / NV_LINES_IN_SLICE;
698 	u32 sl_bit_shift = __builtin_ctz(HTE_SLICE_SIZE);
699 
700 	gs->conf_rval = tegra_hte_readl(gs, HTE_TECTRL);
701 	for (i = 0; i < slices; i++) {
702 		spin_lock(&gs->sl[i].s_lock);
703 		gs->sl[i].r_val = tegra_hte_readl(gs,
704 				((i << sl_bit_shift) + HTE_SLICE0_TETEN));
705 		set_bit(HTE_SUSPEND, &gs->sl[i].flags);
706 		spin_unlock(&gs->sl[i].s_lock);
707 	}
708 
709 	return 0;
710 }
711 
712 static const struct dev_pm_ops tegra_hte_pm = {
713 	SET_LATE_SYSTEM_SLEEP_PM_OPS(tegra_hte_suspend_late,
714 				     tegra_hte_resume_early)
715 };
716 
717 static struct platform_driver tegra_hte_driver = {
718 	.probe = tegra_hte_probe,
719 	.driver = {
720 		.name = "tegra_hte",
721 		.pm = &tegra_hte_pm,
722 		.of_match_table = tegra_hte_of_match,
723 	},
724 };
725 
726 module_platform_driver(tegra_hte_driver);
727 
728 MODULE_AUTHOR("Dipen Patel <dipenp@nvidia.com>");
729 MODULE_DESCRIPTION("NVIDIA Tegra HTE (Hardware Timestamping Engine) driver");
730 MODULE_LICENSE("GPL");
731