162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci// Copyright (c) 2018, The Linux Foundation. All rights reserved.
362306a36Sopenharmony_ci
462306a36Sopenharmony_ci#include <linux/kernel.h>
562306a36Sopenharmony_ci#include <linux/init.h>
662306a36Sopenharmony_ci#include <linux/module.h>
762306a36Sopenharmony_ci#include <linux/platform_device.h>
862306a36Sopenharmony_ci#include <linux/err.h>
962306a36Sopenharmony_ci#include <linux/io.h>
1062306a36Sopenharmony_ci#include <linux/of.h>
1162306a36Sopenharmony_ci#include <linux/of_device.h>
1262306a36Sopenharmony_ci#include <linux/clk.h>
1362306a36Sopenharmony_ci#include <linux/clk-provider.h>
1462306a36Sopenharmony_ci#include <linux/slab.h>
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_ci#include "clk-krait.h"
1762306a36Sopenharmony_ci
1862306a36Sopenharmony_cienum {
1962306a36Sopenharmony_ci	cpu0_mux = 0,
2062306a36Sopenharmony_ci	cpu1_mux,
2162306a36Sopenharmony_ci	cpu2_mux,
2262306a36Sopenharmony_ci	cpu3_mux,
2362306a36Sopenharmony_ci	l2_mux,
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci	clks_max,
2662306a36Sopenharmony_ci};
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic unsigned int sec_mux_map[] = {
2962306a36Sopenharmony_ci	2,
3062306a36Sopenharmony_ci	0,
3162306a36Sopenharmony_ci};
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_cistatic unsigned int pri_mux_map[] = {
3462306a36Sopenharmony_ci	1,
3562306a36Sopenharmony_ci	2,
3662306a36Sopenharmony_ci	0,
3762306a36Sopenharmony_ci};
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci/*
4062306a36Sopenharmony_ci * Notifier function for switching the muxes to safe parent
4162306a36Sopenharmony_ci * while the hfpll is getting reprogrammed.
4262306a36Sopenharmony_ci */
4362306a36Sopenharmony_cistatic int krait_notifier_cb(struct notifier_block *nb,
4462306a36Sopenharmony_ci			     unsigned long event,
4562306a36Sopenharmony_ci			     void *data)
4662306a36Sopenharmony_ci{
4762306a36Sopenharmony_ci	int ret = 0;
4862306a36Sopenharmony_ci	struct krait_mux_clk *mux = container_of(nb, struct krait_mux_clk,
4962306a36Sopenharmony_ci						 clk_nb);
5062306a36Sopenharmony_ci	/* Switch to safe parent */
5162306a36Sopenharmony_ci	if (event == PRE_RATE_CHANGE) {
5262306a36Sopenharmony_ci		mux->old_index = krait_mux_clk_ops.get_parent(&mux->hw);
5362306a36Sopenharmony_ci		ret = krait_mux_clk_ops.set_parent(&mux->hw, mux->safe_sel);
5462306a36Sopenharmony_ci		mux->reparent = false;
5562306a36Sopenharmony_ci	/*
5662306a36Sopenharmony_ci	 * By the time POST_RATE_CHANGE notifier is called,
5762306a36Sopenharmony_ci	 * clk framework itself would have changed the parent for the new rate.
5862306a36Sopenharmony_ci	 * Only otherwise, put back to the old parent.
5962306a36Sopenharmony_ci	 */
6062306a36Sopenharmony_ci	} else if (event == POST_RATE_CHANGE) {
6162306a36Sopenharmony_ci		if (!mux->reparent)
6262306a36Sopenharmony_ci			ret = krait_mux_clk_ops.set_parent(&mux->hw,
6362306a36Sopenharmony_ci							   mux->old_index);
6462306a36Sopenharmony_ci	}
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	return notifier_from_errno(ret);
6762306a36Sopenharmony_ci}
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_cistatic int krait_notifier_register(struct device *dev, struct clk *clk,
7062306a36Sopenharmony_ci				   struct krait_mux_clk *mux)
7162306a36Sopenharmony_ci{
7262306a36Sopenharmony_ci	int ret = 0;
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_ci	mux->clk_nb.notifier_call = krait_notifier_cb;
7562306a36Sopenharmony_ci	ret = devm_clk_notifier_register(dev, clk, &mux->clk_nb);
7662306a36Sopenharmony_ci	if (ret)
7762306a36Sopenharmony_ci		dev_err(dev, "failed to register clock notifier: %d\n", ret);
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	return ret;
8062306a36Sopenharmony_ci}
8162306a36Sopenharmony_ci
8262306a36Sopenharmony_cistatic struct clk_hw *
8362306a36Sopenharmony_cikrait_add_div(struct device *dev, int id, const char *s, unsigned int offset)
8462306a36Sopenharmony_ci{
8562306a36Sopenharmony_ci	struct krait_div2_clk *div;
8662306a36Sopenharmony_ci	static struct clk_parent_data p_data[1];
8762306a36Sopenharmony_ci	struct clk_init_data init = {
8862306a36Sopenharmony_ci		.num_parents = ARRAY_SIZE(p_data),
8962306a36Sopenharmony_ci		.ops = &krait_div2_clk_ops,
9062306a36Sopenharmony_ci		.flags = CLK_SET_RATE_PARENT,
9162306a36Sopenharmony_ci	};
9262306a36Sopenharmony_ci	struct clk_hw *clk;
9362306a36Sopenharmony_ci	char *parent_name;
9462306a36Sopenharmony_ci	int cpu, ret;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	div = devm_kzalloc(dev, sizeof(*div), GFP_KERNEL);
9762306a36Sopenharmony_ci	if (!div)
9862306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci	div->width = 2;
10162306a36Sopenharmony_ci	div->shift = 6;
10262306a36Sopenharmony_ci	div->lpl = id >= 0;
10362306a36Sopenharmony_ci	div->offset = offset;
10462306a36Sopenharmony_ci	div->hw.init = &init;
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	init.name = kasprintf(GFP_KERNEL, "hfpll%s_div", s);
10762306a36Sopenharmony_ci	if (!init.name)
10862306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_ci	init.parent_data = p_data;
11162306a36Sopenharmony_ci	parent_name = kasprintf(GFP_KERNEL, "hfpll%s", s);
11262306a36Sopenharmony_ci	if (!parent_name) {
11362306a36Sopenharmony_ci		clk = ERR_PTR(-ENOMEM);
11462306a36Sopenharmony_ci		goto err_parent_name;
11562306a36Sopenharmony_ci	}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	p_data[0].fw_name = parent_name;
11862306a36Sopenharmony_ci	p_data[0].name = parent_name;
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_ci	ret = devm_clk_hw_register(dev, &div->hw);
12162306a36Sopenharmony_ci	if (ret) {
12262306a36Sopenharmony_ci		clk = ERR_PTR(ret);
12362306a36Sopenharmony_ci		goto err_clk;
12462306a36Sopenharmony_ci	}
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	clk = &div->hw;
12762306a36Sopenharmony_ci
12862306a36Sopenharmony_ci	/* clk-krait ignore any rate change if mux is not flagged as enabled */
12962306a36Sopenharmony_ci	if (id < 0)
13062306a36Sopenharmony_ci		for_each_online_cpu(cpu)
13162306a36Sopenharmony_ci			clk_prepare_enable(div->hw.clk);
13262306a36Sopenharmony_ci	else
13362306a36Sopenharmony_ci		clk_prepare_enable(div->hw.clk);
13462306a36Sopenharmony_ci
13562306a36Sopenharmony_cierr_clk:
13662306a36Sopenharmony_ci	kfree(parent_name);
13762306a36Sopenharmony_cierr_parent_name:
13862306a36Sopenharmony_ci	kfree(init.name);
13962306a36Sopenharmony_ci
14062306a36Sopenharmony_ci	return clk;
14162306a36Sopenharmony_ci}
14262306a36Sopenharmony_ci
14362306a36Sopenharmony_cistatic struct clk_hw *
14462306a36Sopenharmony_cikrait_add_sec_mux(struct device *dev, int id, const char *s,
14562306a36Sopenharmony_ci		  unsigned int offset, bool unique_aux)
14662306a36Sopenharmony_ci{
14762306a36Sopenharmony_ci	int cpu, ret;
14862306a36Sopenharmony_ci	struct krait_mux_clk *mux;
14962306a36Sopenharmony_ci	static struct clk_parent_data sec_mux_list[2] = {
15062306a36Sopenharmony_ci		{ .name = "qsb", .fw_name = "qsb" },
15162306a36Sopenharmony_ci		{},
15262306a36Sopenharmony_ci	};
15362306a36Sopenharmony_ci	struct clk_init_data init = {
15462306a36Sopenharmony_ci		.parent_data = sec_mux_list,
15562306a36Sopenharmony_ci		.num_parents = ARRAY_SIZE(sec_mux_list),
15662306a36Sopenharmony_ci		.ops = &krait_mux_clk_ops,
15762306a36Sopenharmony_ci		.flags = CLK_SET_RATE_PARENT,
15862306a36Sopenharmony_ci	};
15962306a36Sopenharmony_ci	struct clk_hw *clk;
16062306a36Sopenharmony_ci	char *parent_name;
16162306a36Sopenharmony_ci
16262306a36Sopenharmony_ci	mux = devm_kzalloc(dev, sizeof(*mux), GFP_KERNEL);
16362306a36Sopenharmony_ci	if (!mux)
16462306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_ci	mux->offset = offset;
16762306a36Sopenharmony_ci	mux->lpl = id >= 0;
16862306a36Sopenharmony_ci	mux->mask = 0x3;
16962306a36Sopenharmony_ci	mux->shift = 2;
17062306a36Sopenharmony_ci	mux->parent_map = sec_mux_map;
17162306a36Sopenharmony_ci	mux->hw.init = &init;
17262306a36Sopenharmony_ci	mux->safe_sel = 0;
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	/* Checking for qcom,krait-cc-v1 or qcom,krait-cc-v2 is not
17562306a36Sopenharmony_ci	 * enough to limit this to apq/ipq8064. Directly check machine
17662306a36Sopenharmony_ci	 * compatible to correctly handle this errata.
17762306a36Sopenharmony_ci	 */
17862306a36Sopenharmony_ci	if (of_machine_is_compatible("qcom,ipq8064") ||
17962306a36Sopenharmony_ci	    of_machine_is_compatible("qcom,apq8064"))
18062306a36Sopenharmony_ci		mux->disable_sec_src_gating = true;
18162306a36Sopenharmony_ci
18262306a36Sopenharmony_ci	init.name = kasprintf(GFP_KERNEL, "krait%s_sec_mux", s);
18362306a36Sopenharmony_ci	if (!init.name)
18462306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci	if (unique_aux) {
18762306a36Sopenharmony_ci		parent_name = kasprintf(GFP_KERNEL, "acpu%s_aux", s);
18862306a36Sopenharmony_ci		if (!parent_name) {
18962306a36Sopenharmony_ci			clk = ERR_PTR(-ENOMEM);
19062306a36Sopenharmony_ci			goto err_aux;
19162306a36Sopenharmony_ci		}
19262306a36Sopenharmony_ci		sec_mux_list[1].fw_name = parent_name;
19362306a36Sopenharmony_ci		sec_mux_list[1].name = parent_name;
19462306a36Sopenharmony_ci	} else {
19562306a36Sopenharmony_ci		sec_mux_list[1].name = "apu_aux";
19662306a36Sopenharmony_ci	}
19762306a36Sopenharmony_ci
19862306a36Sopenharmony_ci	ret = devm_clk_hw_register(dev, &mux->hw);
19962306a36Sopenharmony_ci	if (ret) {
20062306a36Sopenharmony_ci		clk = ERR_PTR(ret);
20162306a36Sopenharmony_ci		goto err_clk;
20262306a36Sopenharmony_ci	}
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_ci	clk = &mux->hw;
20562306a36Sopenharmony_ci
20662306a36Sopenharmony_ci	ret = krait_notifier_register(dev, mux->hw.clk, mux);
20762306a36Sopenharmony_ci	if (ret) {
20862306a36Sopenharmony_ci		clk = ERR_PTR(ret);
20962306a36Sopenharmony_ci		goto err_clk;
21062306a36Sopenharmony_ci	}
21162306a36Sopenharmony_ci
21262306a36Sopenharmony_ci	/* clk-krait ignore any rate change if mux is not flagged as enabled */
21362306a36Sopenharmony_ci	if (id < 0)
21462306a36Sopenharmony_ci		for_each_online_cpu(cpu)
21562306a36Sopenharmony_ci			clk_prepare_enable(mux->hw.clk);
21662306a36Sopenharmony_ci	else
21762306a36Sopenharmony_ci		clk_prepare_enable(mux->hw.clk);
21862306a36Sopenharmony_ci
21962306a36Sopenharmony_cierr_clk:
22062306a36Sopenharmony_ci	if (unique_aux)
22162306a36Sopenharmony_ci		kfree(parent_name);
22262306a36Sopenharmony_cierr_aux:
22362306a36Sopenharmony_ci	kfree(init.name);
22462306a36Sopenharmony_ci	return clk;
22562306a36Sopenharmony_ci}
22662306a36Sopenharmony_ci
22762306a36Sopenharmony_cistatic struct clk_hw *
22862306a36Sopenharmony_cikrait_add_pri_mux(struct device *dev, struct clk_hw *hfpll_div, struct clk_hw *sec_mux,
22962306a36Sopenharmony_ci		  int id, const char *s, unsigned int offset)
23062306a36Sopenharmony_ci{
23162306a36Sopenharmony_ci	int ret;
23262306a36Sopenharmony_ci	struct krait_mux_clk *mux;
23362306a36Sopenharmony_ci	static struct clk_parent_data p_data[3];
23462306a36Sopenharmony_ci	struct clk_init_data init = {
23562306a36Sopenharmony_ci		.parent_data = p_data,
23662306a36Sopenharmony_ci		.num_parents = ARRAY_SIZE(p_data),
23762306a36Sopenharmony_ci		.ops = &krait_mux_clk_ops,
23862306a36Sopenharmony_ci		.flags = CLK_SET_RATE_PARENT,
23962306a36Sopenharmony_ci	};
24062306a36Sopenharmony_ci	struct clk_hw *clk;
24162306a36Sopenharmony_ci	char *hfpll_name;
24262306a36Sopenharmony_ci
24362306a36Sopenharmony_ci	mux = devm_kzalloc(dev, sizeof(*mux), GFP_KERNEL);
24462306a36Sopenharmony_ci	if (!mux)
24562306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
24662306a36Sopenharmony_ci
24762306a36Sopenharmony_ci	mux->mask = 0x3;
24862306a36Sopenharmony_ci	mux->shift = 0;
24962306a36Sopenharmony_ci	mux->offset = offset;
25062306a36Sopenharmony_ci	mux->lpl = id >= 0;
25162306a36Sopenharmony_ci	mux->parent_map = pri_mux_map;
25262306a36Sopenharmony_ci	mux->hw.init = &init;
25362306a36Sopenharmony_ci	mux->safe_sel = 2;
25462306a36Sopenharmony_ci
25562306a36Sopenharmony_ci	init.name = kasprintf(GFP_KERNEL, "krait%s_pri_mux", s);
25662306a36Sopenharmony_ci	if (!init.name)
25762306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
25862306a36Sopenharmony_ci
25962306a36Sopenharmony_ci	hfpll_name = kasprintf(GFP_KERNEL, "hfpll%s", s);
26062306a36Sopenharmony_ci	if (!hfpll_name) {
26162306a36Sopenharmony_ci		clk = ERR_PTR(-ENOMEM);
26262306a36Sopenharmony_ci		goto err_hfpll;
26362306a36Sopenharmony_ci	}
26462306a36Sopenharmony_ci
26562306a36Sopenharmony_ci	p_data[0].fw_name = hfpll_name;
26662306a36Sopenharmony_ci	p_data[0].name = hfpll_name;
26762306a36Sopenharmony_ci
26862306a36Sopenharmony_ci	p_data[1].hw = hfpll_div;
26962306a36Sopenharmony_ci	p_data[2].hw = sec_mux;
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ci	ret = devm_clk_hw_register(dev, &mux->hw);
27262306a36Sopenharmony_ci	if (ret) {
27362306a36Sopenharmony_ci		clk = ERR_PTR(ret);
27462306a36Sopenharmony_ci		goto err_clk;
27562306a36Sopenharmony_ci	}
27662306a36Sopenharmony_ci
27762306a36Sopenharmony_ci	clk = &mux->hw;
27862306a36Sopenharmony_ci
27962306a36Sopenharmony_ci	ret = krait_notifier_register(dev, mux->hw.clk, mux);
28062306a36Sopenharmony_ci	if (ret)
28162306a36Sopenharmony_ci		clk = ERR_PTR(ret);
28262306a36Sopenharmony_ci
28362306a36Sopenharmony_cierr_clk:
28462306a36Sopenharmony_ci	kfree(hfpll_name);
28562306a36Sopenharmony_cierr_hfpll:
28662306a36Sopenharmony_ci	kfree(init.name);
28762306a36Sopenharmony_ci	return clk;
28862306a36Sopenharmony_ci}
28962306a36Sopenharmony_ci
29062306a36Sopenharmony_ci/* id < 0 for L2, otherwise id == physical CPU number */
29162306a36Sopenharmony_cistatic struct clk_hw *krait_add_clks(struct device *dev, int id, bool unique_aux)
29262306a36Sopenharmony_ci{
29362306a36Sopenharmony_ci	struct clk_hw *hfpll_div, *sec_mux, *pri_mux;
29462306a36Sopenharmony_ci	unsigned int offset;
29562306a36Sopenharmony_ci	void *p = NULL;
29662306a36Sopenharmony_ci	const char *s;
29762306a36Sopenharmony_ci
29862306a36Sopenharmony_ci	if (id >= 0) {
29962306a36Sopenharmony_ci		offset = 0x4501 + (0x1000 * id);
30062306a36Sopenharmony_ci		s = p = kasprintf(GFP_KERNEL, "%d", id);
30162306a36Sopenharmony_ci		if (!s)
30262306a36Sopenharmony_ci			return ERR_PTR(-ENOMEM);
30362306a36Sopenharmony_ci	} else {
30462306a36Sopenharmony_ci		offset = 0x500;
30562306a36Sopenharmony_ci		s = "_l2";
30662306a36Sopenharmony_ci	}
30762306a36Sopenharmony_ci
30862306a36Sopenharmony_ci	hfpll_div = krait_add_div(dev, id, s, offset);
30962306a36Sopenharmony_ci	if (IS_ERR(hfpll_div)) {
31062306a36Sopenharmony_ci		pri_mux = hfpll_div;
31162306a36Sopenharmony_ci		goto err;
31262306a36Sopenharmony_ci	}
31362306a36Sopenharmony_ci
31462306a36Sopenharmony_ci	sec_mux = krait_add_sec_mux(dev, id, s, offset, unique_aux);
31562306a36Sopenharmony_ci	if (IS_ERR(sec_mux)) {
31662306a36Sopenharmony_ci		pri_mux = sec_mux;
31762306a36Sopenharmony_ci		goto err;
31862306a36Sopenharmony_ci	}
31962306a36Sopenharmony_ci
32062306a36Sopenharmony_ci	pri_mux = krait_add_pri_mux(dev, hfpll_div, sec_mux, id, s, offset);
32162306a36Sopenharmony_ci
32262306a36Sopenharmony_cierr:
32362306a36Sopenharmony_ci	kfree(p);
32462306a36Sopenharmony_ci	return pri_mux;
32562306a36Sopenharmony_ci}
32662306a36Sopenharmony_ci
32762306a36Sopenharmony_cistatic struct clk *krait_of_get(struct of_phandle_args *clkspec, void *data)
32862306a36Sopenharmony_ci{
32962306a36Sopenharmony_ci	unsigned int idx = clkspec->args[0];
33062306a36Sopenharmony_ci	struct clk **clks = data;
33162306a36Sopenharmony_ci
33262306a36Sopenharmony_ci	if (idx >= clks_max) {
33362306a36Sopenharmony_ci		pr_err("%s: invalid clock index %d\n", __func__, idx);
33462306a36Sopenharmony_ci		return ERR_PTR(-EINVAL);
33562306a36Sopenharmony_ci	}
33662306a36Sopenharmony_ci
33762306a36Sopenharmony_ci	return clks[idx] ? : ERR_PTR(-ENODEV);
33862306a36Sopenharmony_ci}
33962306a36Sopenharmony_ci
34062306a36Sopenharmony_cistatic const struct of_device_id krait_cc_match_table[] = {
34162306a36Sopenharmony_ci	{ .compatible = "qcom,krait-cc-v1", (void *)1UL },
34262306a36Sopenharmony_ci	{ .compatible = "qcom,krait-cc-v2" },
34362306a36Sopenharmony_ci	{}
34462306a36Sopenharmony_ci};
34562306a36Sopenharmony_ciMODULE_DEVICE_TABLE(of, krait_cc_match_table);
34662306a36Sopenharmony_ci
34762306a36Sopenharmony_cistatic int krait_cc_probe(struct platform_device *pdev)
34862306a36Sopenharmony_ci{
34962306a36Sopenharmony_ci	struct device *dev = &pdev->dev;
35062306a36Sopenharmony_ci	const struct of_device_id *id;
35162306a36Sopenharmony_ci	unsigned long cur_rate, aux_rate;
35262306a36Sopenharmony_ci	int cpu;
35362306a36Sopenharmony_ci	struct clk_hw *mux, *l2_pri_mux;
35462306a36Sopenharmony_ci	struct clk *clk, **clks;
35562306a36Sopenharmony_ci
35662306a36Sopenharmony_ci	id = of_match_device(krait_cc_match_table, dev);
35762306a36Sopenharmony_ci	if (!id)
35862306a36Sopenharmony_ci		return -ENODEV;
35962306a36Sopenharmony_ci
36062306a36Sopenharmony_ci	/* Rate is 1 because 0 causes problems for __clk_mux_determine_rate */
36162306a36Sopenharmony_ci	clk = clk_register_fixed_rate(dev, "qsb", NULL, 0, 1);
36262306a36Sopenharmony_ci	if (IS_ERR(clk))
36362306a36Sopenharmony_ci		return PTR_ERR(clk);
36462306a36Sopenharmony_ci
36562306a36Sopenharmony_ci	if (!id->data) {
36662306a36Sopenharmony_ci		clk = clk_register_fixed_factor(dev, "acpu_aux",
36762306a36Sopenharmony_ci						"gpll0_vote", 0, 1, 2);
36862306a36Sopenharmony_ci		if (IS_ERR(clk))
36962306a36Sopenharmony_ci			return PTR_ERR(clk);
37062306a36Sopenharmony_ci	}
37162306a36Sopenharmony_ci
37262306a36Sopenharmony_ci	/* Krait configurations have at most 4 CPUs and one L2 */
37362306a36Sopenharmony_ci	clks = devm_kcalloc(dev, clks_max, sizeof(*clks), GFP_KERNEL);
37462306a36Sopenharmony_ci	if (!clks)
37562306a36Sopenharmony_ci		return -ENOMEM;
37662306a36Sopenharmony_ci
37762306a36Sopenharmony_ci	for_each_possible_cpu(cpu) {
37862306a36Sopenharmony_ci		mux = krait_add_clks(dev, cpu, id->data);
37962306a36Sopenharmony_ci		if (IS_ERR(mux))
38062306a36Sopenharmony_ci			return PTR_ERR(mux);
38162306a36Sopenharmony_ci		clks[cpu] = mux->clk;
38262306a36Sopenharmony_ci	}
38362306a36Sopenharmony_ci
38462306a36Sopenharmony_ci	l2_pri_mux = krait_add_clks(dev, -1, id->data);
38562306a36Sopenharmony_ci	if (IS_ERR(l2_pri_mux))
38662306a36Sopenharmony_ci		return PTR_ERR(l2_pri_mux);
38762306a36Sopenharmony_ci	clks[l2_mux] = l2_pri_mux->clk;
38862306a36Sopenharmony_ci
38962306a36Sopenharmony_ci	/*
39062306a36Sopenharmony_ci	 * We don't want the CPU or L2 clocks to be turned off at late init
39162306a36Sopenharmony_ci	 * if CPUFREQ or HOTPLUG configs are disabled. So, bump up the
39262306a36Sopenharmony_ci	 * refcount of these clocks. Any cpufreq/hotplug manager can assume
39362306a36Sopenharmony_ci	 * that the clocks have already been prepared and enabled by the time
39462306a36Sopenharmony_ci	 * they take over.
39562306a36Sopenharmony_ci	 */
39662306a36Sopenharmony_ci	for_each_online_cpu(cpu) {
39762306a36Sopenharmony_ci		clk_prepare_enable(clks[l2_mux]);
39862306a36Sopenharmony_ci		WARN(clk_prepare_enable(clks[cpu]),
39962306a36Sopenharmony_ci		     "Unable to turn on CPU%d clock", cpu);
40062306a36Sopenharmony_ci	}
40162306a36Sopenharmony_ci
40262306a36Sopenharmony_ci	/*
40362306a36Sopenharmony_ci	 * Force reinit of HFPLLs and muxes to overwrite any potential
40462306a36Sopenharmony_ci	 * incorrect configuration of HFPLLs and muxes by the bootloader.
40562306a36Sopenharmony_ci	 * While at it, also make sure the cores are running at known rates
40662306a36Sopenharmony_ci	 * and print the current rate.
40762306a36Sopenharmony_ci	 *
40862306a36Sopenharmony_ci	 * The clocks are set to aux clock rate first to make sure the
40962306a36Sopenharmony_ci	 * secondary mux is not sourcing off of QSB. The rate is then set to
41062306a36Sopenharmony_ci	 * two different rates to force a HFPLL reinit under all
41162306a36Sopenharmony_ci	 * circumstances.
41262306a36Sopenharmony_ci	 */
41362306a36Sopenharmony_ci	cur_rate = clk_get_rate(clks[l2_mux]);
41462306a36Sopenharmony_ci	aux_rate = 384000000;
41562306a36Sopenharmony_ci	if (cur_rate < aux_rate) {
41662306a36Sopenharmony_ci		pr_info("L2 @ Undefined rate. Forcing new rate.\n");
41762306a36Sopenharmony_ci		cur_rate = aux_rate;
41862306a36Sopenharmony_ci	}
41962306a36Sopenharmony_ci	clk_set_rate(clks[l2_mux], aux_rate);
42062306a36Sopenharmony_ci	clk_set_rate(clks[l2_mux], 2);
42162306a36Sopenharmony_ci	clk_set_rate(clks[l2_mux], cur_rate);
42262306a36Sopenharmony_ci	pr_info("L2 @ %lu KHz\n", clk_get_rate(clks[l2_mux]) / 1000);
42362306a36Sopenharmony_ci	for_each_possible_cpu(cpu) {
42462306a36Sopenharmony_ci		clk = clks[cpu];
42562306a36Sopenharmony_ci		cur_rate = clk_get_rate(clk);
42662306a36Sopenharmony_ci		if (cur_rate < aux_rate) {
42762306a36Sopenharmony_ci			pr_info("CPU%d @ Undefined rate. Forcing new rate.\n", cpu);
42862306a36Sopenharmony_ci			cur_rate = aux_rate;
42962306a36Sopenharmony_ci		}
43062306a36Sopenharmony_ci
43162306a36Sopenharmony_ci		clk_set_rate(clk, aux_rate);
43262306a36Sopenharmony_ci		clk_set_rate(clk, 2);
43362306a36Sopenharmony_ci		clk_set_rate(clk, cur_rate);
43462306a36Sopenharmony_ci		pr_info("CPU%d @ %lu KHz\n", cpu, clk_get_rate(clk) / 1000);
43562306a36Sopenharmony_ci	}
43662306a36Sopenharmony_ci
43762306a36Sopenharmony_ci	of_clk_add_provider(dev->of_node, krait_of_get, clks);
43862306a36Sopenharmony_ci
43962306a36Sopenharmony_ci	return 0;
44062306a36Sopenharmony_ci}
44162306a36Sopenharmony_ci
44262306a36Sopenharmony_cistatic struct platform_driver krait_cc_driver = {
44362306a36Sopenharmony_ci	.probe = krait_cc_probe,
44462306a36Sopenharmony_ci	.driver = {
44562306a36Sopenharmony_ci		.name = "krait-cc",
44662306a36Sopenharmony_ci		.of_match_table = krait_cc_match_table,
44762306a36Sopenharmony_ci	},
44862306a36Sopenharmony_ci};
44962306a36Sopenharmony_cimodule_platform_driver(krait_cc_driver);
45062306a36Sopenharmony_ci
45162306a36Sopenharmony_ciMODULE_DESCRIPTION("Krait CPU Clock Driver");
45262306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
45362306a36Sopenharmony_ciMODULE_ALIAS("platform:krait-cc");
454