xref: /linux/drivers/clk/tegra/clk-bpmp.c (revision 3a2c4d55e32ad65efebdb6de44eef3bfa08bb49d)
1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Copyright (C) 2016-2022 NVIDIA Corporation
4  */
5 
6 #include <linux/clk-provider.h>
7 #include <linux/device.h>
8 #include <linux/seq_buf.h>
9 #include <linux/slab.h>
10 
11 #include <soc/tegra/bpmp.h>
12 #include <soc/tegra/bpmp-abi.h>
13 
14 #define TEGRA_BPMP_DUMP_CLOCK_INFO	0
15 
16 #define TEGRA_BPMP_CLK_HAS_MUX		BIT(0)
17 #define TEGRA_BPMP_CLK_HAS_SET_RATE	BIT(1)
18 #define TEGRA_BPMP_CLK_IS_ROOT		BIT(2)
19 
20 struct tegra_bpmp_clk_info {
21 	unsigned int id;
22 	char name[MRQ_CLK_NAME_MAXLEN];
23 	unsigned int parents[MRQ_CLK_MAX_PARENTS];
24 	unsigned int num_parents;
25 	unsigned long flags;
26 };
27 
28 struct tegra_bpmp_clk {
29 	struct clk_hw hw;
30 
31 	struct tegra_bpmp *bpmp;
32 	unsigned int id;
33 
34 	unsigned int num_parents;
35 	unsigned int *parents;
36 };
37 
38 static inline struct tegra_bpmp_clk *to_tegra_bpmp_clk(struct clk_hw *hw)
39 {
40 	return container_of(hw, struct tegra_bpmp_clk, hw);
41 }
42 
43 struct tegra_bpmp_clk_message {
44 	unsigned int cmd;
45 	unsigned int id;
46 
47 	struct {
48 		const void *data;
49 		size_t size;
50 	} tx;
51 
52 	struct {
53 		void *data;
54 		size_t size;
55 		int ret;
56 	} rx;
57 };
58 
59 static int tegra_bpmp_clk_transfer(struct tegra_bpmp *bpmp,
60 				   const struct tegra_bpmp_clk_message *clk)
61 {
62 	struct mrq_clk_request request;
63 	struct tegra_bpmp_message msg;
64 	void *req = &request;
65 	int err;
66 
67 	memset(&request, 0, sizeof(request));
68 	request.cmd_and_id = (clk->cmd << 24) | clk->id;
69 
70 	/*
71 	 * The mrq_clk_request structure has an anonymous union at offset 4
72 	 * that contains all possible sub-command structures. Copy the data
73 	 * to that union. Ideally we'd be able to refer to it by name, but
74 	 * doing so would require changing the ABI header and increase the
75 	 * maintenance burden.
76 	 */
77 	memcpy(req + 4, clk->tx.data, clk->tx.size);
78 
79 	memset(&msg, 0, sizeof(msg));
80 	msg.mrq = MRQ_CLK;
81 	msg.tx.data = &request;
82 	msg.tx.size = sizeof(request);
83 	msg.rx.data = clk->rx.data;
84 	msg.rx.size = clk->rx.size;
85 
86 	err = tegra_bpmp_transfer(bpmp, &msg);
87 	if (err < 0)
88 		return err;
89 	else if (msg.rx.ret < 0)
90 		return -EINVAL;
91 
92 	return 0;
93 }
94 
95 static int tegra_bpmp_clk_prepare(struct clk_hw *hw)
96 {
97 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
98 	struct tegra_bpmp_clk_message msg;
99 
100 	memset(&msg, 0, sizeof(msg));
101 	msg.cmd = CMD_CLK_ENABLE;
102 	msg.id = clk->id;
103 
104 	return tegra_bpmp_clk_transfer(clk->bpmp, &msg);
105 }
106 
107 static void tegra_bpmp_clk_unprepare(struct clk_hw *hw)
108 {
109 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
110 	struct tegra_bpmp_clk_message msg;
111 	int err;
112 
113 	memset(&msg, 0, sizeof(msg));
114 	msg.cmd = CMD_CLK_DISABLE;
115 	msg.id = clk->id;
116 
117 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
118 	if (err < 0)
119 		dev_err(clk->bpmp->dev, "failed to disable clock %s: %d\n",
120 			clk_hw_get_name(hw), err);
121 }
122 
123 static int tegra_bpmp_clk_is_prepared(struct clk_hw *hw)
124 {
125 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
126 	struct cmd_clk_is_enabled_response response;
127 	struct tegra_bpmp_clk_message msg;
128 	int err;
129 
130 	memset(&msg, 0, sizeof(msg));
131 	msg.cmd = CMD_CLK_IS_ENABLED;
132 	msg.id = clk->id;
133 	msg.rx.data = &response;
134 	msg.rx.size = sizeof(response);
135 
136 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
137 	if (err < 0)
138 		return err;
139 
140 	return response.state;
141 }
142 
143 static unsigned long tegra_bpmp_clk_recalc_rate(struct clk_hw *hw,
144 						unsigned long parent_rate)
145 {
146 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
147 	struct cmd_clk_get_rate_response response;
148 	struct cmd_clk_get_rate_request request;
149 	struct tegra_bpmp_clk_message msg;
150 	int err;
151 
152 	memset(&msg, 0, sizeof(msg));
153 	msg.cmd = CMD_CLK_GET_RATE;
154 	msg.id = clk->id;
155 	msg.tx.data = &request;
156 	msg.tx.size = sizeof(request);
157 	msg.rx.data = &response;
158 	msg.rx.size = sizeof(response);
159 
160 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
161 	if (err < 0)
162 		return 0;
163 
164 	return response.rate;
165 }
166 
167 static int tegra_bpmp_clk_determine_rate(struct clk_hw *hw,
168 					 struct clk_rate_request *rate_req)
169 {
170 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
171 	struct cmd_clk_round_rate_response response;
172 	struct cmd_clk_round_rate_request request;
173 	struct tegra_bpmp_clk_message msg;
174 	unsigned long rate;
175 	int err;
176 
177 	rate = clamp(rate_req->rate, rate_req->min_rate, rate_req->max_rate);
178 
179 	memset(&request, 0, sizeof(request));
180 	request.rate = min_t(u64, rate, S64_MAX);
181 
182 	memset(&msg, 0, sizeof(msg));
183 	msg.cmd = CMD_CLK_ROUND_RATE;
184 	msg.id = clk->id;
185 	msg.tx.data = &request;
186 	msg.tx.size = sizeof(request);
187 	msg.rx.data = &response;
188 	msg.rx.size = sizeof(response);
189 
190 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
191 	if (err < 0)
192 		return err;
193 
194 	rate_req->rate = (unsigned long)response.rate;
195 
196 	return 0;
197 }
198 
199 static int tegra_bpmp_clk_set_parent(struct clk_hw *hw, u8 index)
200 {
201 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
202 	struct cmd_clk_set_parent_response response;
203 	struct cmd_clk_set_parent_request request;
204 	struct tegra_bpmp_clk_message msg;
205 	int err;
206 
207 	memset(&request, 0, sizeof(request));
208 	request.parent_id = clk->parents[index];
209 
210 	memset(&msg, 0, sizeof(msg));
211 	msg.cmd = CMD_CLK_SET_PARENT;
212 	msg.id = clk->id;
213 	msg.tx.data = &request;
214 	msg.tx.size = sizeof(request);
215 	msg.rx.data = &response;
216 	msg.rx.size = sizeof(response);
217 
218 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
219 	if (err < 0)
220 		return err;
221 
222 	/* XXX check parent ID in response */
223 
224 	return 0;
225 }
226 
227 static u8 tegra_bpmp_clk_get_parent(struct clk_hw *hw)
228 {
229 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
230 	struct cmd_clk_get_parent_response response;
231 	struct tegra_bpmp_clk_message msg;
232 	unsigned int i;
233 	int err;
234 
235 	memset(&msg, 0, sizeof(msg));
236 	msg.cmd = CMD_CLK_GET_PARENT;
237 	msg.id = clk->id;
238 	msg.rx.data = &response;
239 	msg.rx.size = sizeof(response);
240 
241 	err = tegra_bpmp_clk_transfer(clk->bpmp, &msg);
242 	if (err < 0) {
243 		dev_err(clk->bpmp->dev, "failed to get parent for %s: %d\n",
244 			clk_hw_get_name(hw), err);
245 		return U8_MAX;
246 	}
247 
248 	for (i = 0; i < clk->num_parents; i++)
249 		if (clk->parents[i] == response.parent_id)
250 			return i;
251 
252 	return U8_MAX;
253 }
254 
255 static int tegra_bpmp_clk_set_rate(struct clk_hw *hw, unsigned long rate,
256 				   unsigned long parent_rate)
257 {
258 	struct tegra_bpmp_clk *clk = to_tegra_bpmp_clk(hw);
259 	struct cmd_clk_set_rate_response response;
260 	struct cmd_clk_set_rate_request request;
261 	struct tegra_bpmp_clk_message msg;
262 
263 	memset(&request, 0, sizeof(request));
264 	request.rate = min_t(u64, rate, S64_MAX);
265 
266 	memset(&msg, 0, sizeof(msg));
267 	msg.cmd = CMD_CLK_SET_RATE;
268 	msg.id = clk->id;
269 	msg.tx.data = &request;
270 	msg.tx.size = sizeof(request);
271 	msg.rx.data = &response;
272 	msg.rx.size = sizeof(response);
273 
274 	return tegra_bpmp_clk_transfer(clk->bpmp, &msg);
275 }
276 
277 static const struct clk_ops tegra_bpmp_clk_gate_ops = {
278 	.prepare = tegra_bpmp_clk_prepare,
279 	.unprepare = tegra_bpmp_clk_unprepare,
280 	.is_prepared = tegra_bpmp_clk_is_prepared,
281 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
282 };
283 
284 static const struct clk_ops tegra_bpmp_clk_mux_ops = {
285 	.prepare = tegra_bpmp_clk_prepare,
286 	.unprepare = tegra_bpmp_clk_unprepare,
287 	.is_prepared = tegra_bpmp_clk_is_prepared,
288 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
289 	.determine_rate = clk_hw_determine_rate_no_reparent,
290 	.set_parent = tegra_bpmp_clk_set_parent,
291 	.get_parent = tegra_bpmp_clk_get_parent,
292 };
293 
294 static const struct clk_ops tegra_bpmp_clk_rate_ops = {
295 	.prepare = tegra_bpmp_clk_prepare,
296 	.unprepare = tegra_bpmp_clk_unprepare,
297 	.is_prepared = tegra_bpmp_clk_is_prepared,
298 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
299 	.determine_rate = tegra_bpmp_clk_determine_rate,
300 	.set_rate = tegra_bpmp_clk_set_rate,
301 };
302 
303 static const struct clk_ops tegra_bpmp_clk_mux_rate_ops = {
304 	.prepare = tegra_bpmp_clk_prepare,
305 	.unprepare = tegra_bpmp_clk_unprepare,
306 	.is_prepared = tegra_bpmp_clk_is_prepared,
307 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
308 	.determine_rate = tegra_bpmp_clk_determine_rate,
309 	.set_parent = tegra_bpmp_clk_set_parent,
310 	.get_parent = tegra_bpmp_clk_get_parent,
311 	.set_rate = tegra_bpmp_clk_set_rate,
312 };
313 
314 static const struct clk_ops tegra_bpmp_clk_mux_read_only_ops = {
315 	.get_parent = tegra_bpmp_clk_get_parent,
316 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
317 };
318 
319 static const struct clk_ops tegra_bpmp_clk_read_only_ops = {
320 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
321 };
322 
323 static const struct clk_ops tegra_bpmp_clk_gate_mux_read_only_ops = {
324 	.prepare = tegra_bpmp_clk_prepare,
325 	.unprepare = tegra_bpmp_clk_unprepare,
326 	.is_prepared = tegra_bpmp_clk_is_prepared,
327 	.recalc_rate = tegra_bpmp_clk_recalc_rate,
328 	.get_parent = tegra_bpmp_clk_get_parent,
329 };
330 
331 static int tegra_bpmp_clk_get_max_id(struct tegra_bpmp *bpmp)
332 {
333 	struct cmd_clk_get_max_clk_id_response response;
334 	struct tegra_bpmp_clk_message msg;
335 	int err;
336 
337 	memset(&msg, 0, sizeof(msg));
338 	msg.cmd = CMD_CLK_GET_MAX_CLK_ID;
339 	msg.rx.data = &response;
340 	msg.rx.size = sizeof(response);
341 
342 	err = tegra_bpmp_clk_transfer(bpmp, &msg);
343 	if (err < 0)
344 		return err;
345 
346 	if (response.max_id > INT_MAX)
347 		return -E2BIG;
348 
349 	return response.max_id;
350 }
351 
352 static int tegra_bpmp_clk_get_info(struct tegra_bpmp *bpmp, unsigned int id,
353 				   struct tegra_bpmp_clk_info *info)
354 {
355 	struct cmd_clk_get_all_info_response response;
356 	struct tegra_bpmp_clk_message msg;
357 	unsigned int i;
358 	int err;
359 
360 	memset(&msg, 0, sizeof(msg));
361 	msg.cmd = CMD_CLK_GET_ALL_INFO;
362 	msg.id = id;
363 	msg.rx.data = &response;
364 	msg.rx.size = sizeof(response);
365 
366 	err = tegra_bpmp_clk_transfer(bpmp, &msg);
367 	if (err < 0)
368 		return err;
369 
370 	if (dev_to_node(bpmp->dev) == NUMA_NO_NODE) {
371 		strscpy(info->name, response.name, sizeof(info->name));
372 	} else {
373 		err = snprintf(info->name, sizeof(info->name), "%d-%s",
374 			       dev_to_node(bpmp->dev), response.name);
375 		if (WARN_ON(err >= sizeof(info->name)))
376 			return -E2BIG;
377 	}
378 
379 	info->num_parents = response.num_parents;
380 
381 	for (i = 0; i < info->num_parents; i++)
382 		info->parents[i] = response.parents[i];
383 
384 	info->flags = response.flags;
385 
386 	return 0;
387 }
388 
389 static void tegra_bpmp_clk_info_dump(struct tegra_bpmp *bpmp,
390 				     const char *level,
391 				     const struct tegra_bpmp_clk_info *info)
392 {
393 	const char *prefix = "";
394 	struct seq_buf buf;
395 	unsigned int i;
396 	char flags[64];
397 
398 	seq_buf_init(&buf, flags, sizeof(flags));
399 
400 	if (info->flags)
401 		seq_buf_printf(&buf, "(");
402 
403 	if (info->flags & TEGRA_BPMP_CLK_HAS_MUX) {
404 		seq_buf_printf(&buf, "%smux", prefix);
405 		prefix = ", ";
406 	}
407 
408 	if ((info->flags & TEGRA_BPMP_CLK_HAS_SET_RATE) == 0) {
409 		seq_buf_printf(&buf, "%sfixed", prefix);
410 		prefix = ", ";
411 	}
412 
413 	if (info->flags & TEGRA_BPMP_CLK_IS_ROOT) {
414 		seq_buf_printf(&buf, "%sroot", prefix);
415 		prefix = ", ";
416 	}
417 
418 	if (info->flags)
419 		seq_buf_printf(&buf, ")");
420 
421 	dev_printk(level, bpmp->dev, "%03u: %s\n", info->id, info->name);
422 	dev_printk(level, bpmp->dev, "  flags: %lx %s\n", info->flags, flags);
423 	dev_printk(level, bpmp->dev, "  parents: %u\n", info->num_parents);
424 
425 	for (i = 0; i < info->num_parents; i++)
426 		dev_printk(level, bpmp->dev, "    %03u\n", info->parents[i]);
427 }
428 
429 static int tegra_bpmp_probe_clocks(struct tegra_bpmp *bpmp,
430 				   struct tegra_bpmp_clk_info **clocksp)
431 {
432 	struct tegra_bpmp_clk_info *clocks;
433 	unsigned int max_id, id, count = 0;
434 	unsigned int holes = 0;
435 	int err;
436 
437 	err = tegra_bpmp_clk_get_max_id(bpmp);
438 	if (err < 0)
439 		return err;
440 
441 	max_id = err;
442 
443 	dev_dbg(bpmp->dev, "maximum clock ID: %u\n", max_id);
444 
445 	clocks = kzalloc_objs(*clocks, max_id + 1);
446 	if (!clocks)
447 		return -ENOMEM;
448 
449 	for (id = 0; id <= max_id; id++) {
450 		struct tegra_bpmp_clk_info *info = &clocks[count];
451 
452 		err = tegra_bpmp_clk_get_info(bpmp, id, info);
453 		if (err < 0)
454 			continue;
455 
456 		if (info->num_parents >= U8_MAX) {
457 			dev_err(bpmp->dev,
458 				"clock %u has too many parents (%u, max: %u)\n",
459 				id, info->num_parents, U8_MAX);
460 			continue;
461 		}
462 
463 		/* clock not exposed by BPMP */
464 		if (info->name[0] == '\0') {
465 			holes++;
466 			continue;
467 		}
468 
469 		info->id = id;
470 		count++;
471 
472 		if (TEGRA_BPMP_DUMP_CLOCK_INFO)
473 			tegra_bpmp_clk_info_dump(bpmp, KERN_DEBUG, info);
474 	}
475 
476 	dev_dbg(bpmp->dev, "holes: %u\n", holes);
477 	*clocksp = clocks;
478 
479 	return count;
480 }
481 
482 static unsigned int
483 tegra_bpmp_clk_id_to_index(const struct tegra_bpmp_clk_info *clocks,
484 			   unsigned int num_clocks, unsigned int id)
485 {
486 	unsigned int i;
487 
488 	for (i = 0; i < num_clocks; i++)
489 		if (clocks[i].id == id)
490 			return i;
491 
492 	return UINT_MAX;
493 }
494 
495 static const struct tegra_bpmp_clk_info *
496 tegra_bpmp_clk_find(const struct tegra_bpmp_clk_info *clocks,
497 		    unsigned int num_clocks, unsigned int id)
498 {
499 	unsigned int i;
500 
501 	i = tegra_bpmp_clk_id_to_index(clocks, num_clocks, id);
502 
503 	if (i < num_clocks)
504 		return &clocks[i];
505 
506 	return NULL;
507 }
508 
509 static struct tegra_bpmp_clk *
510 tegra_bpmp_clk_register(struct tegra_bpmp *bpmp,
511 			const struct tegra_bpmp_clk_info *info,
512 			const struct tegra_bpmp_clk_info *clocks,
513 			unsigned int num_clocks)
514 {
515 	struct tegra_bpmp_clk *clk;
516 	struct clk_init_data init;
517 	const char **parents;
518 	unsigned int i;
519 	int err;
520 
521 	clk = devm_kzalloc(bpmp->dev, sizeof(*clk), GFP_KERNEL);
522 	if (!clk)
523 		return ERR_PTR(-ENOMEM);
524 
525 	clk->id = info->id;
526 	clk->bpmp = bpmp;
527 
528 	clk->parents = devm_kcalloc(bpmp->dev, info->num_parents,
529 				    sizeof(*clk->parents), GFP_KERNEL);
530 	if (!clk->parents)
531 		return ERR_PTR(-ENOMEM);
532 
533 	clk->num_parents = info->num_parents;
534 
535 	/* hardware clock initialization */
536 	memset(&init, 0, sizeof(init));
537 	init.name = info->name;
538 	clk->hw.init = &init;
539 	if (info->flags & BPMP_CLK_STATE_CHANGE_DENIED) {
540 		if ((info->flags & BPMP_CLK_RATE_PARENT_CHANGE_DENIED) == 0) {
541 			dev_WARN(bpmp->dev,
542 				"Firmware bug! Inconsistent permission bits for clock %s. State and parent/rate changes disabled.",
543 				 init.name);
544 		}
545 		if (info->flags & TEGRA_BPMP_CLK_HAS_MUX)
546 			init.ops = &tegra_bpmp_clk_mux_read_only_ops;
547 		else
548 			init.ops = &tegra_bpmp_clk_read_only_ops;
549 	} else if (info->flags & BPMP_CLK_RATE_PARENT_CHANGE_DENIED) {
550 		if (info->flags & TEGRA_BPMP_CLK_HAS_MUX)
551 			init.ops = &tegra_bpmp_clk_gate_mux_read_only_ops;
552 		else
553 			init.ops = &tegra_bpmp_clk_gate_ops;
554 	} else if (info->flags & TEGRA_BPMP_CLK_HAS_MUX) {
555 		if (info->flags & TEGRA_BPMP_CLK_HAS_SET_RATE)
556 			init.ops = &tegra_bpmp_clk_mux_rate_ops;
557 		else
558 			init.ops = &tegra_bpmp_clk_mux_ops;
559 	} else {
560 		if (info->flags & TEGRA_BPMP_CLK_HAS_SET_RATE)
561 			init.ops = &tegra_bpmp_clk_rate_ops;
562 		else
563 			init.ops = &tegra_bpmp_clk_gate_ops;
564 	}
565 
566 	init.num_parents = info->num_parents;
567 
568 	parents = kcalloc(info->num_parents, sizeof(*parents), GFP_KERNEL);
569 	if (!parents)
570 		return ERR_PTR(-ENOMEM);
571 
572 	for (i = 0; i < info->num_parents; i++) {
573 		const struct tegra_bpmp_clk_info *parent;
574 
575 		/* keep a private copy of the ID to parent index map */
576 		clk->parents[i] = info->parents[i];
577 
578 		parent = tegra_bpmp_clk_find(clocks, num_clocks,
579 					     info->parents[i]);
580 		if (!parent) {
581 			dev_err(bpmp->dev, "no parent %u found for %u\n",
582 				info->parents[i], info->id);
583 			continue;
584 		}
585 
586 		parents[i] = parent->name;
587 	}
588 
589 	init.parent_names = parents;
590 
591 	err = devm_clk_hw_register(bpmp->dev, &clk->hw);
592 
593 	kfree(parents);
594 
595 	if (err < 0)
596 		return ERR_PTR(err);
597 
598 	return clk;
599 }
600 
601 static void tegra_bpmp_register_clocks_one(struct tegra_bpmp *bpmp,
602 					   struct tegra_bpmp_clk_info *infos,
603 					   unsigned int i,
604 					   unsigned int count)
605 {
606 	unsigned int j;
607 	struct tegra_bpmp_clk_info *info;
608 	struct tegra_bpmp_clk *clk;
609 
610 	if (bpmp->clocks[i]) {
611 		/* already registered */
612 		return;
613 	}
614 
615 	info = &infos[i];
616 	for (j = 0; j < info->num_parents; ++j) {
617 		unsigned int p_id = info->parents[j];
618 		unsigned int p_i = tegra_bpmp_clk_id_to_index(infos, count,
619 							      p_id);
620 		if (p_i < count)
621 			tegra_bpmp_register_clocks_one(bpmp, infos, p_i, count);
622 	}
623 
624 	clk = tegra_bpmp_clk_register(bpmp, info, infos, count);
625 	if (IS_ERR(clk)) {
626 		dev_err(bpmp->dev,
627 			"failed to register clock %u (%s): %ld\n",
628 			info->id, info->name, PTR_ERR(clk));
629 		/* intentionally store the error pointer to
630 		 * bpmp->clocks[i] to avoid re-attempting the
631 		 * registration later
632 		 */
633 	}
634 
635 	bpmp->clocks[i] = clk;
636 }
637 
638 static int tegra_bpmp_register_clocks(struct tegra_bpmp *bpmp,
639 				      struct tegra_bpmp_clk_info *infos,
640 				      unsigned int count)
641 {
642 	unsigned int i;
643 
644 	bpmp->num_clocks = count;
645 
646 	bpmp->clocks = devm_kcalloc(bpmp->dev, count, sizeof(*bpmp->clocks), GFP_KERNEL);
647 	if (!bpmp->clocks)
648 		return -ENOMEM;
649 
650 	for (i = 0; i < count; i++) {
651 		tegra_bpmp_register_clocks_one(bpmp, infos, i, count);
652 	}
653 
654 	return 0;
655 }
656 
657 static void tegra_bpmp_unregister_clocks(struct tegra_bpmp *bpmp)
658 {
659 	unsigned int i;
660 
661 	for (i = 0; i < bpmp->num_clocks; i++)
662 		clk_hw_unregister(&bpmp->clocks[i]->hw);
663 }
664 
665 static struct clk_hw *tegra_bpmp_clk_of_xlate(struct of_phandle_args *clkspec,
666 					      void *data)
667 {
668 	unsigned int id = clkspec->args[0], i;
669 	struct tegra_bpmp *bpmp = data;
670 
671 	for (i = 0; i < bpmp->num_clocks; i++) {
672 		struct tegra_bpmp_clk *clk = bpmp->clocks[i];
673 
674 		if (!clk)
675 			continue;
676 
677 		if (clk->id == id)
678 			return &clk->hw;
679 	}
680 
681 	return NULL;
682 }
683 
684 int tegra_bpmp_init_clocks(struct tegra_bpmp *bpmp)
685 {
686 	struct tegra_bpmp_clk_info *clocks;
687 	unsigned int count;
688 	int err;
689 
690 	err = tegra_bpmp_probe_clocks(bpmp, &clocks);
691 	if (err < 0)
692 		return err;
693 
694 	count = err;
695 
696 	dev_dbg(bpmp->dev, "%u clocks probed\n", count);
697 
698 	err = tegra_bpmp_register_clocks(bpmp, clocks, count);
699 	if (err < 0)
700 		goto free;
701 
702 	err = of_clk_add_hw_provider(bpmp->dev->of_node,
703 				     tegra_bpmp_clk_of_xlate,
704 				     bpmp);
705 	if (err < 0) {
706 		tegra_bpmp_unregister_clocks(bpmp);
707 		goto free;
708 	}
709 
710 free:
711 	kfree(clocks);
712 	return err;
713 }
714