162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0+
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Voltage regulators coupler for NVIDIA Tegra30
462306a36Sopenharmony_ci * Copyright (C) 2019 GRATE-DRIVER project
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci * Voltage constraints borrowed from downstream kernel sources
762306a36Sopenharmony_ci * Copyright (C) 2010-2011 NVIDIA Corporation
862306a36Sopenharmony_ci */
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#define pr_fmt(fmt)	"tegra voltage-coupler: " fmt
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_ci#include <linux/init.h>
1362306a36Sopenharmony_ci#include <linux/kernel.h>
1462306a36Sopenharmony_ci#include <linux/of.h>
1562306a36Sopenharmony_ci#include <linux/reboot.h>
1662306a36Sopenharmony_ci#include <linux/regulator/coupler.h>
1762306a36Sopenharmony_ci#include <linux/regulator/driver.h>
1862306a36Sopenharmony_ci#include <linux/regulator/machine.h>
1962306a36Sopenharmony_ci#include <linux/suspend.h>
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci#include <soc/tegra/fuse.h>
2262306a36Sopenharmony_ci#include <soc/tegra/pmc.h>
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistruct tegra_regulator_coupler {
2562306a36Sopenharmony_ci	struct regulator_coupler coupler;
2662306a36Sopenharmony_ci	struct regulator_dev *core_rdev;
2762306a36Sopenharmony_ci	struct regulator_dev *cpu_rdev;
2862306a36Sopenharmony_ci	struct notifier_block reboot_notifier;
2962306a36Sopenharmony_ci	struct notifier_block suspend_notifier;
3062306a36Sopenharmony_ci	int core_min_uV, cpu_min_uV;
3162306a36Sopenharmony_ci	bool sys_reboot_mode_req;
3262306a36Sopenharmony_ci	bool sys_reboot_mode;
3362306a36Sopenharmony_ci	bool sys_suspend_mode_req;
3462306a36Sopenharmony_ci	bool sys_suspend_mode;
3562306a36Sopenharmony_ci};
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_cistatic inline struct tegra_regulator_coupler *
3862306a36Sopenharmony_cito_tegra_coupler(struct regulator_coupler *coupler)
3962306a36Sopenharmony_ci{
4062306a36Sopenharmony_ci	return container_of(coupler, struct tegra_regulator_coupler, coupler);
4162306a36Sopenharmony_ci}
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_cistatic int tegra30_core_limit(struct tegra_regulator_coupler *tegra,
4462306a36Sopenharmony_ci			      struct regulator_dev *core_rdev)
4562306a36Sopenharmony_ci{
4662306a36Sopenharmony_ci	int core_min_uV = 0;
4762306a36Sopenharmony_ci	int core_max_uV;
4862306a36Sopenharmony_ci	int core_cur_uV;
4962306a36Sopenharmony_ci	int err;
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci	/*
5262306a36Sopenharmony_ci	 * Tegra30 SoC has critical DVFS-capable devices that are
5362306a36Sopenharmony_ci	 * permanently-active or active at a boot time, like EMC
5462306a36Sopenharmony_ci	 * (DRAM controller) or Display controller for example.
5562306a36Sopenharmony_ci	 *
5662306a36Sopenharmony_ci	 * The voltage of a CORE SoC power domain shall not be dropped below
5762306a36Sopenharmony_ci	 * a minimum level, which is determined by device's clock rate.
5862306a36Sopenharmony_ci	 * This means that we can't fully allow CORE voltage scaling until
5962306a36Sopenharmony_ci	 * the state of all DVFS-critical CORE devices is synced.
6062306a36Sopenharmony_ci	 */
6162306a36Sopenharmony_ci	if (tegra_pmc_core_domain_state_synced() && !tegra->sys_reboot_mode) {
6262306a36Sopenharmony_ci		pr_info_once("voltage state synced\n");
6362306a36Sopenharmony_ci		return 0;
6462306a36Sopenharmony_ci	}
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	if (tegra->core_min_uV > 0)
6762306a36Sopenharmony_ci		return tegra->core_min_uV;
6862306a36Sopenharmony_ci
6962306a36Sopenharmony_ci	core_cur_uV = regulator_get_voltage_rdev(core_rdev);
7062306a36Sopenharmony_ci	if (core_cur_uV < 0)
7162306a36Sopenharmony_ci		return core_cur_uV;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	core_max_uV = max(core_cur_uV, 1200000);
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	err = regulator_check_voltage(core_rdev, &core_min_uV, &core_max_uV);
7662306a36Sopenharmony_ci	if (err)
7762306a36Sopenharmony_ci		return err;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	/*
8062306a36Sopenharmony_ci	 * Limit minimum CORE voltage to a value left from bootloader or,
8162306a36Sopenharmony_ci	 * if it's unreasonably low value, to the most common 1.2v or to
8262306a36Sopenharmony_ci	 * whatever maximum value defined via board's device-tree.
8362306a36Sopenharmony_ci	 */
8462306a36Sopenharmony_ci	tegra->core_min_uV = core_max_uV;
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ci	pr_info("core voltage initialized to %duV\n", tegra->core_min_uV);
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	return tegra->core_min_uV;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_cistatic int tegra30_core_cpu_limit(int cpu_uV)
9262306a36Sopenharmony_ci{
9362306a36Sopenharmony_ci	if (cpu_uV < 800000)
9462306a36Sopenharmony_ci		return 950000;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	if (cpu_uV < 900000)
9762306a36Sopenharmony_ci		return 1000000;
9862306a36Sopenharmony_ci
9962306a36Sopenharmony_ci	if (cpu_uV < 1000000)
10062306a36Sopenharmony_ci		return 1100000;
10162306a36Sopenharmony_ci
10262306a36Sopenharmony_ci	if (cpu_uV < 1100000)
10362306a36Sopenharmony_ci		return 1200000;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	if (cpu_uV < 1250000) {
10662306a36Sopenharmony_ci		switch (tegra_sku_info.cpu_speedo_id) {
10762306a36Sopenharmony_ci		case 0 ... 1:
10862306a36Sopenharmony_ci		case 4:
10962306a36Sopenharmony_ci		case 7 ... 8:
11062306a36Sopenharmony_ci			return 1200000;
11162306a36Sopenharmony_ci
11262306a36Sopenharmony_ci		default:
11362306a36Sopenharmony_ci			return 1300000;
11462306a36Sopenharmony_ci		}
11562306a36Sopenharmony_ci	}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_ci	return -EINVAL;
11862306a36Sopenharmony_ci}
11962306a36Sopenharmony_ci
12062306a36Sopenharmony_cistatic int tegra30_cpu_nominal_uV(void)
12162306a36Sopenharmony_ci{
12262306a36Sopenharmony_ci	switch (tegra_sku_info.cpu_speedo_id) {
12362306a36Sopenharmony_ci	case 10 ... 11:
12462306a36Sopenharmony_ci		return  850000;
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	case 9:
12762306a36Sopenharmony_ci		return  912000;
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_ci	case 1 ...  3:
13062306a36Sopenharmony_ci	case 7 ...  8:
13162306a36Sopenharmony_ci		return 1050000;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	default:
13462306a36Sopenharmony_ci		return 1125000;
13562306a36Sopenharmony_ci
13662306a36Sopenharmony_ci	case  4 ...  6:
13762306a36Sopenharmony_ci	case 12 ... 13:
13862306a36Sopenharmony_ci		return 1237000;
13962306a36Sopenharmony_ci	}
14062306a36Sopenharmony_ci}
14162306a36Sopenharmony_ci
14262306a36Sopenharmony_cistatic int tegra30_core_nominal_uV(void)
14362306a36Sopenharmony_ci{
14462306a36Sopenharmony_ci	switch (tegra_sku_info.soc_speedo_id) {
14562306a36Sopenharmony_ci	case 0:
14662306a36Sopenharmony_ci		return 1200000;
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_ci	case 1:
14962306a36Sopenharmony_ci		if (tegra_sku_info.cpu_speedo_id != 7 &&
15062306a36Sopenharmony_ci		    tegra_sku_info.cpu_speedo_id != 8)
15162306a36Sopenharmony_ci			return 1200000;
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_ci		fallthrough;
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	case 2:
15662306a36Sopenharmony_ci		if (tegra_sku_info.cpu_speedo_id != 13)
15762306a36Sopenharmony_ci			return 1300000;
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci		return 1350000;
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci	default:
16262306a36Sopenharmony_ci		return 1250000;
16362306a36Sopenharmony_ci	}
16462306a36Sopenharmony_ci}
16562306a36Sopenharmony_ci
16662306a36Sopenharmony_cistatic int tegra30_voltage_update(struct tegra_regulator_coupler *tegra,
16762306a36Sopenharmony_ci				  struct regulator_dev *cpu_rdev,
16862306a36Sopenharmony_ci				  struct regulator_dev *core_rdev)
16962306a36Sopenharmony_ci{
17062306a36Sopenharmony_ci	int core_min_uV, core_max_uV = INT_MAX;
17162306a36Sopenharmony_ci	int cpu_min_uV, cpu_max_uV = INT_MAX;
17262306a36Sopenharmony_ci	int cpu_min_uV_consumers = 0;
17362306a36Sopenharmony_ci	int core_min_limited_uV;
17462306a36Sopenharmony_ci	int core_target_uV;
17562306a36Sopenharmony_ci	int cpu_target_uV;
17662306a36Sopenharmony_ci	int core_max_step;
17762306a36Sopenharmony_ci	int cpu_max_step;
17862306a36Sopenharmony_ci	int max_spread;
17962306a36Sopenharmony_ci	int core_uV;
18062306a36Sopenharmony_ci	int cpu_uV;
18162306a36Sopenharmony_ci	int err;
18262306a36Sopenharmony_ci
18362306a36Sopenharmony_ci	/*
18462306a36Sopenharmony_ci	 * CPU voltage should not got lower than 300mV from the CORE.
18562306a36Sopenharmony_ci	 * CPU voltage should stay below the CORE by 100mV+, depending
18662306a36Sopenharmony_ci	 * by the CORE voltage. This applies to all Tegra30 SoC's.
18762306a36Sopenharmony_ci	 */
18862306a36Sopenharmony_ci	max_spread = cpu_rdev->constraints->max_spread[0];
18962306a36Sopenharmony_ci	cpu_max_step = cpu_rdev->constraints->max_uV_step;
19062306a36Sopenharmony_ci	core_max_step = core_rdev->constraints->max_uV_step;
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	if (!max_spread) {
19362306a36Sopenharmony_ci		pr_err_once("cpu-core max-spread is undefined in device-tree\n");
19462306a36Sopenharmony_ci		max_spread = 300000;
19562306a36Sopenharmony_ci	}
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_ci	if (!cpu_max_step) {
19862306a36Sopenharmony_ci		pr_err_once("cpu max-step is undefined in device-tree\n");
19962306a36Sopenharmony_ci		cpu_max_step = 150000;
20062306a36Sopenharmony_ci	}
20162306a36Sopenharmony_ci
20262306a36Sopenharmony_ci	if (!core_max_step) {
20362306a36Sopenharmony_ci		pr_err_once("core max-step is undefined in device-tree\n");
20462306a36Sopenharmony_ci		core_max_step = 150000;
20562306a36Sopenharmony_ci	}
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	/*
20862306a36Sopenharmony_ci	 * The CORE voltage scaling is currently not hooked up in drivers,
20962306a36Sopenharmony_ci	 * hence we will limit the minimum CORE voltage to a reasonable value.
21062306a36Sopenharmony_ci	 * This should be good enough for the time being.
21162306a36Sopenharmony_ci	 */
21262306a36Sopenharmony_ci	core_min_uV = tegra30_core_limit(tegra, core_rdev);
21362306a36Sopenharmony_ci	if (core_min_uV < 0)
21462306a36Sopenharmony_ci		return core_min_uV;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	err = regulator_check_consumers(core_rdev, &core_min_uV, &core_max_uV,
21762306a36Sopenharmony_ci					PM_SUSPEND_ON);
21862306a36Sopenharmony_ci	if (err)
21962306a36Sopenharmony_ci		return err;
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_ci	/* prepare voltage level for suspend */
22262306a36Sopenharmony_ci	if (tegra->sys_suspend_mode)
22362306a36Sopenharmony_ci		core_min_uV = clamp(tegra30_core_nominal_uV(),
22462306a36Sopenharmony_ci				    core_min_uV, core_max_uV);
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	core_uV = regulator_get_voltage_rdev(core_rdev);
22762306a36Sopenharmony_ci	if (core_uV < 0)
22862306a36Sopenharmony_ci		return core_uV;
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ci	cpu_min_uV = core_min_uV - max_spread;
23162306a36Sopenharmony_ci
23262306a36Sopenharmony_ci	err = regulator_check_consumers(cpu_rdev, &cpu_min_uV, &cpu_max_uV,
23362306a36Sopenharmony_ci					PM_SUSPEND_ON);
23462306a36Sopenharmony_ci	if (err)
23562306a36Sopenharmony_ci		return err;
23662306a36Sopenharmony_ci
23762306a36Sopenharmony_ci	err = regulator_check_consumers(cpu_rdev, &cpu_min_uV_consumers,
23862306a36Sopenharmony_ci					&cpu_max_uV, PM_SUSPEND_ON);
23962306a36Sopenharmony_ci	if (err)
24062306a36Sopenharmony_ci		return err;
24162306a36Sopenharmony_ci
24262306a36Sopenharmony_ci	err = regulator_check_voltage(cpu_rdev, &cpu_min_uV, &cpu_max_uV);
24362306a36Sopenharmony_ci	if (err)
24462306a36Sopenharmony_ci		return err;
24562306a36Sopenharmony_ci
24662306a36Sopenharmony_ci	cpu_uV = regulator_get_voltage_rdev(cpu_rdev);
24762306a36Sopenharmony_ci	if (cpu_uV < 0)
24862306a36Sopenharmony_ci		return cpu_uV;
24962306a36Sopenharmony_ci
25062306a36Sopenharmony_ci	/* store boot voltage level */
25162306a36Sopenharmony_ci	if (!tegra->cpu_min_uV)
25262306a36Sopenharmony_ci		tegra->cpu_min_uV = cpu_uV;
25362306a36Sopenharmony_ci
25462306a36Sopenharmony_ci	/*
25562306a36Sopenharmony_ci	 * CPU's regulator may not have any consumers, hence the voltage
25662306a36Sopenharmony_ci	 * must not be changed in that case because CPU simply won't
25762306a36Sopenharmony_ci	 * survive the voltage drop if it's running on a higher frequency.
25862306a36Sopenharmony_ci	 */
25962306a36Sopenharmony_ci	if (!cpu_min_uV_consumers)
26062306a36Sopenharmony_ci		cpu_min_uV = max(cpu_uV, cpu_min_uV);
26162306a36Sopenharmony_ci
26262306a36Sopenharmony_ci	/*
26362306a36Sopenharmony_ci	 * Bootloader shall set up voltages correctly, but if it
26462306a36Sopenharmony_ci	 * happens that there is a violation, then try to fix it
26562306a36Sopenharmony_ci	 * at first.
26662306a36Sopenharmony_ci	 */
26762306a36Sopenharmony_ci	core_min_limited_uV = tegra30_core_cpu_limit(cpu_uV);
26862306a36Sopenharmony_ci	if (core_min_limited_uV < 0)
26962306a36Sopenharmony_ci		return core_min_limited_uV;
27062306a36Sopenharmony_ci
27162306a36Sopenharmony_ci	core_min_uV = max(core_min_uV, tegra30_core_cpu_limit(cpu_min_uV));
27262306a36Sopenharmony_ci
27362306a36Sopenharmony_ci	err = regulator_check_voltage(core_rdev, &core_min_uV, &core_max_uV);
27462306a36Sopenharmony_ci	if (err)
27562306a36Sopenharmony_ci		return err;
27662306a36Sopenharmony_ci
27762306a36Sopenharmony_ci	/* restore boot voltage level */
27862306a36Sopenharmony_ci	if (tegra->sys_reboot_mode)
27962306a36Sopenharmony_ci		cpu_min_uV = max(cpu_min_uV, tegra->cpu_min_uV);
28062306a36Sopenharmony_ci
28162306a36Sopenharmony_ci	/* prepare voltage level for suspend */
28262306a36Sopenharmony_ci	if (tegra->sys_suspend_mode)
28362306a36Sopenharmony_ci		cpu_min_uV = clamp(tegra30_cpu_nominal_uV(),
28462306a36Sopenharmony_ci				   cpu_min_uV, cpu_max_uV);
28562306a36Sopenharmony_ci
28662306a36Sopenharmony_ci	if (core_min_limited_uV > core_uV) {
28762306a36Sopenharmony_ci		pr_err("core voltage constraint violated: %d %d %d\n",
28862306a36Sopenharmony_ci		       core_uV, core_min_limited_uV, cpu_uV);
28962306a36Sopenharmony_ci		goto update_core;
29062306a36Sopenharmony_ci	}
29162306a36Sopenharmony_ci
29262306a36Sopenharmony_ci	while (cpu_uV != cpu_min_uV || core_uV != core_min_uV) {
29362306a36Sopenharmony_ci		if (cpu_uV < cpu_min_uV) {
29462306a36Sopenharmony_ci			cpu_target_uV = min(cpu_uV + cpu_max_step, cpu_min_uV);
29562306a36Sopenharmony_ci		} else {
29662306a36Sopenharmony_ci			cpu_target_uV = max(cpu_uV - cpu_max_step, cpu_min_uV);
29762306a36Sopenharmony_ci			cpu_target_uV = max(core_uV - max_spread, cpu_target_uV);
29862306a36Sopenharmony_ci		}
29962306a36Sopenharmony_ci
30062306a36Sopenharmony_ci		if (cpu_uV == cpu_target_uV)
30162306a36Sopenharmony_ci			goto update_core;
30262306a36Sopenharmony_ci
30362306a36Sopenharmony_ci		err = regulator_set_voltage_rdev(cpu_rdev,
30462306a36Sopenharmony_ci						 cpu_target_uV,
30562306a36Sopenharmony_ci						 cpu_max_uV,
30662306a36Sopenharmony_ci						 PM_SUSPEND_ON);
30762306a36Sopenharmony_ci		if (err)
30862306a36Sopenharmony_ci			return err;
30962306a36Sopenharmony_ci
31062306a36Sopenharmony_ci		cpu_uV = cpu_target_uV;
31162306a36Sopenharmony_ciupdate_core:
31262306a36Sopenharmony_ci		core_min_limited_uV = tegra30_core_cpu_limit(cpu_uV);
31362306a36Sopenharmony_ci		if (core_min_limited_uV < 0)
31462306a36Sopenharmony_ci			return core_min_limited_uV;
31562306a36Sopenharmony_ci
31662306a36Sopenharmony_ci		core_target_uV = max(core_min_limited_uV, core_min_uV);
31762306a36Sopenharmony_ci
31862306a36Sopenharmony_ci		if (core_uV < core_target_uV) {
31962306a36Sopenharmony_ci			core_target_uV = min(core_target_uV, core_uV + core_max_step);
32062306a36Sopenharmony_ci			core_target_uV = min(core_target_uV, cpu_uV + max_spread);
32162306a36Sopenharmony_ci		} else {
32262306a36Sopenharmony_ci			core_target_uV = max(core_target_uV, core_uV - core_max_step);
32362306a36Sopenharmony_ci		}
32462306a36Sopenharmony_ci
32562306a36Sopenharmony_ci		if (core_uV == core_target_uV)
32662306a36Sopenharmony_ci			continue;
32762306a36Sopenharmony_ci
32862306a36Sopenharmony_ci		err = regulator_set_voltage_rdev(core_rdev,
32962306a36Sopenharmony_ci						 core_target_uV,
33062306a36Sopenharmony_ci						 core_max_uV,
33162306a36Sopenharmony_ci						 PM_SUSPEND_ON);
33262306a36Sopenharmony_ci		if (err)
33362306a36Sopenharmony_ci			return err;
33462306a36Sopenharmony_ci
33562306a36Sopenharmony_ci		core_uV = core_target_uV;
33662306a36Sopenharmony_ci	}
33762306a36Sopenharmony_ci
33862306a36Sopenharmony_ci	return 0;
33962306a36Sopenharmony_ci}
34062306a36Sopenharmony_ci
34162306a36Sopenharmony_cistatic int tegra30_regulator_balance_voltage(struct regulator_coupler *coupler,
34262306a36Sopenharmony_ci					     struct regulator_dev *rdev,
34362306a36Sopenharmony_ci					     suspend_state_t state)
34462306a36Sopenharmony_ci{
34562306a36Sopenharmony_ci	struct tegra_regulator_coupler *tegra = to_tegra_coupler(coupler);
34662306a36Sopenharmony_ci	struct regulator_dev *core_rdev = tegra->core_rdev;
34762306a36Sopenharmony_ci	struct regulator_dev *cpu_rdev = tegra->cpu_rdev;
34862306a36Sopenharmony_ci
34962306a36Sopenharmony_ci	if ((core_rdev != rdev && cpu_rdev != rdev) || state != PM_SUSPEND_ON) {
35062306a36Sopenharmony_ci		pr_err("regulators are not coupled properly\n");
35162306a36Sopenharmony_ci		return -EINVAL;
35262306a36Sopenharmony_ci	}
35362306a36Sopenharmony_ci
35462306a36Sopenharmony_ci	tegra->sys_reboot_mode = READ_ONCE(tegra->sys_reboot_mode_req);
35562306a36Sopenharmony_ci	tegra->sys_suspend_mode = READ_ONCE(tegra->sys_suspend_mode_req);
35662306a36Sopenharmony_ci
35762306a36Sopenharmony_ci	return tegra30_voltage_update(tegra, cpu_rdev, core_rdev);
35862306a36Sopenharmony_ci}
35962306a36Sopenharmony_ci
36062306a36Sopenharmony_cistatic int tegra30_regulator_prepare_suspend(struct tegra_regulator_coupler *tegra,
36162306a36Sopenharmony_ci					     bool sys_suspend_mode)
36262306a36Sopenharmony_ci{
36362306a36Sopenharmony_ci	int err;
36462306a36Sopenharmony_ci
36562306a36Sopenharmony_ci	if (!tegra->core_rdev || !tegra->cpu_rdev)
36662306a36Sopenharmony_ci		return 0;
36762306a36Sopenharmony_ci
36862306a36Sopenharmony_ci	/*
36962306a36Sopenharmony_ci	 * All power domains are enabled early during resume from suspend
37062306a36Sopenharmony_ci	 * by GENPD core.  Domains like VENC may require a higher voltage
37162306a36Sopenharmony_ci	 * when enabled during resume from suspend.  This also prepares
37262306a36Sopenharmony_ci	 * hardware for resuming from LP0.
37362306a36Sopenharmony_ci	 */
37462306a36Sopenharmony_ci
37562306a36Sopenharmony_ci	WRITE_ONCE(tegra->sys_suspend_mode_req, sys_suspend_mode);
37662306a36Sopenharmony_ci
37762306a36Sopenharmony_ci	err = regulator_sync_voltage_rdev(tegra->cpu_rdev);
37862306a36Sopenharmony_ci	if (err)
37962306a36Sopenharmony_ci		return err;
38062306a36Sopenharmony_ci
38162306a36Sopenharmony_ci	err = regulator_sync_voltage_rdev(tegra->core_rdev);
38262306a36Sopenharmony_ci	if (err)
38362306a36Sopenharmony_ci		return err;
38462306a36Sopenharmony_ci
38562306a36Sopenharmony_ci	return 0;
38662306a36Sopenharmony_ci}
38762306a36Sopenharmony_ci
38862306a36Sopenharmony_cistatic int tegra30_regulator_suspend(struct notifier_block *notifier,
38962306a36Sopenharmony_ci				     unsigned long mode, void *arg)
39062306a36Sopenharmony_ci{
39162306a36Sopenharmony_ci	struct tegra_regulator_coupler *tegra;
39262306a36Sopenharmony_ci	int ret = 0;
39362306a36Sopenharmony_ci
39462306a36Sopenharmony_ci	tegra = container_of(notifier, struct tegra_regulator_coupler,
39562306a36Sopenharmony_ci			     suspend_notifier);
39662306a36Sopenharmony_ci
39762306a36Sopenharmony_ci	switch (mode) {
39862306a36Sopenharmony_ci	case PM_HIBERNATION_PREPARE:
39962306a36Sopenharmony_ci	case PM_RESTORE_PREPARE:
40062306a36Sopenharmony_ci	case PM_SUSPEND_PREPARE:
40162306a36Sopenharmony_ci		ret = tegra30_regulator_prepare_suspend(tegra, true);
40262306a36Sopenharmony_ci		break;
40362306a36Sopenharmony_ci
40462306a36Sopenharmony_ci	case PM_POST_HIBERNATION:
40562306a36Sopenharmony_ci	case PM_POST_RESTORE:
40662306a36Sopenharmony_ci	case PM_POST_SUSPEND:
40762306a36Sopenharmony_ci		ret = tegra30_regulator_prepare_suspend(tegra, false);
40862306a36Sopenharmony_ci		break;
40962306a36Sopenharmony_ci	}
41062306a36Sopenharmony_ci
41162306a36Sopenharmony_ci	if (ret)
41262306a36Sopenharmony_ci		pr_err("failed to prepare regulators: %d\n", ret);
41362306a36Sopenharmony_ci
41462306a36Sopenharmony_ci	return notifier_from_errno(ret);
41562306a36Sopenharmony_ci}
41662306a36Sopenharmony_ci
41762306a36Sopenharmony_cistatic int tegra30_regulator_prepare_reboot(struct tegra_regulator_coupler *tegra,
41862306a36Sopenharmony_ci					    bool sys_reboot_mode)
41962306a36Sopenharmony_ci{
42062306a36Sopenharmony_ci	int err;
42162306a36Sopenharmony_ci
42262306a36Sopenharmony_ci	if (!tegra->core_rdev || !tegra->cpu_rdev)
42362306a36Sopenharmony_ci		return 0;
42462306a36Sopenharmony_ci
42562306a36Sopenharmony_ci	WRITE_ONCE(tegra->sys_reboot_mode_req, true);
42662306a36Sopenharmony_ci
42762306a36Sopenharmony_ci	/*
42862306a36Sopenharmony_ci	 * Some devices use CPU soft-reboot method and in this case we
42962306a36Sopenharmony_ci	 * should ensure that voltages are sane for the reboot by restoring
43062306a36Sopenharmony_ci	 * the minimum boot levels.
43162306a36Sopenharmony_ci	 */
43262306a36Sopenharmony_ci	err = regulator_sync_voltage_rdev(tegra->cpu_rdev);
43362306a36Sopenharmony_ci	if (err)
43462306a36Sopenharmony_ci		return err;
43562306a36Sopenharmony_ci
43662306a36Sopenharmony_ci	err = regulator_sync_voltage_rdev(tegra->core_rdev);
43762306a36Sopenharmony_ci	if (err)
43862306a36Sopenharmony_ci		return err;
43962306a36Sopenharmony_ci
44062306a36Sopenharmony_ci	WRITE_ONCE(tegra->sys_reboot_mode_req, sys_reboot_mode);
44162306a36Sopenharmony_ci
44262306a36Sopenharmony_ci	return 0;
44362306a36Sopenharmony_ci}
44462306a36Sopenharmony_ci
44562306a36Sopenharmony_cistatic int tegra30_regulator_reboot(struct notifier_block *notifier,
44662306a36Sopenharmony_ci				    unsigned long event, void *cmd)
44762306a36Sopenharmony_ci{
44862306a36Sopenharmony_ci	struct tegra_regulator_coupler *tegra;
44962306a36Sopenharmony_ci	int ret;
45062306a36Sopenharmony_ci
45162306a36Sopenharmony_ci	if (event != SYS_RESTART)
45262306a36Sopenharmony_ci		return NOTIFY_DONE;
45362306a36Sopenharmony_ci
45462306a36Sopenharmony_ci	tegra = container_of(notifier, struct tegra_regulator_coupler,
45562306a36Sopenharmony_ci			     reboot_notifier);
45662306a36Sopenharmony_ci
45762306a36Sopenharmony_ci	ret = tegra30_regulator_prepare_reboot(tegra, true);
45862306a36Sopenharmony_ci
45962306a36Sopenharmony_ci	return notifier_from_errno(ret);
46062306a36Sopenharmony_ci}
46162306a36Sopenharmony_ci
46262306a36Sopenharmony_cistatic int tegra30_regulator_attach(struct regulator_coupler *coupler,
46362306a36Sopenharmony_ci				    struct regulator_dev *rdev)
46462306a36Sopenharmony_ci{
46562306a36Sopenharmony_ci	struct tegra_regulator_coupler *tegra = to_tegra_coupler(coupler);
46662306a36Sopenharmony_ci	struct device_node *np = rdev->dev.of_node;
46762306a36Sopenharmony_ci
46862306a36Sopenharmony_ci	if (of_property_read_bool(np, "nvidia,tegra-core-regulator") &&
46962306a36Sopenharmony_ci	    !tegra->core_rdev) {
47062306a36Sopenharmony_ci		tegra->core_rdev = rdev;
47162306a36Sopenharmony_ci		return 0;
47262306a36Sopenharmony_ci	}
47362306a36Sopenharmony_ci
47462306a36Sopenharmony_ci	if (of_property_read_bool(np, "nvidia,tegra-cpu-regulator") &&
47562306a36Sopenharmony_ci	    !tegra->cpu_rdev) {
47662306a36Sopenharmony_ci		tegra->cpu_rdev = rdev;
47762306a36Sopenharmony_ci		return 0;
47862306a36Sopenharmony_ci	}
47962306a36Sopenharmony_ci
48062306a36Sopenharmony_ci	return -EINVAL;
48162306a36Sopenharmony_ci}
48262306a36Sopenharmony_ci
48362306a36Sopenharmony_cistatic int tegra30_regulator_detach(struct regulator_coupler *coupler,
48462306a36Sopenharmony_ci				    struct regulator_dev *rdev)
48562306a36Sopenharmony_ci{
48662306a36Sopenharmony_ci	struct tegra_regulator_coupler *tegra = to_tegra_coupler(coupler);
48762306a36Sopenharmony_ci
48862306a36Sopenharmony_ci	/*
48962306a36Sopenharmony_ci	 * We don't expect regulators to be decoupled during reboot,
49062306a36Sopenharmony_ci	 * this may race with the reboot handler and shouldn't ever
49162306a36Sopenharmony_ci	 * happen in practice.
49262306a36Sopenharmony_ci	 */
49362306a36Sopenharmony_ci	if (WARN_ON_ONCE(system_state > SYSTEM_RUNNING))
49462306a36Sopenharmony_ci		return -EPERM;
49562306a36Sopenharmony_ci
49662306a36Sopenharmony_ci	if (tegra->core_rdev == rdev) {
49762306a36Sopenharmony_ci		tegra->core_rdev = NULL;
49862306a36Sopenharmony_ci		return 0;
49962306a36Sopenharmony_ci	}
50062306a36Sopenharmony_ci
50162306a36Sopenharmony_ci	if (tegra->cpu_rdev == rdev) {
50262306a36Sopenharmony_ci		tegra->cpu_rdev = NULL;
50362306a36Sopenharmony_ci		return 0;
50462306a36Sopenharmony_ci	}
50562306a36Sopenharmony_ci
50662306a36Sopenharmony_ci	return -EINVAL;
50762306a36Sopenharmony_ci}
50862306a36Sopenharmony_ci
50962306a36Sopenharmony_cistatic struct tegra_regulator_coupler tegra30_coupler = {
51062306a36Sopenharmony_ci	.coupler = {
51162306a36Sopenharmony_ci		.attach_regulator = tegra30_regulator_attach,
51262306a36Sopenharmony_ci		.detach_regulator = tegra30_regulator_detach,
51362306a36Sopenharmony_ci		.balance_voltage = tegra30_regulator_balance_voltage,
51462306a36Sopenharmony_ci	},
51562306a36Sopenharmony_ci	.reboot_notifier.notifier_call = tegra30_regulator_reboot,
51662306a36Sopenharmony_ci	.suspend_notifier.notifier_call = tegra30_regulator_suspend,
51762306a36Sopenharmony_ci};
51862306a36Sopenharmony_ci
51962306a36Sopenharmony_cistatic int __init tegra_regulator_coupler_init(void)
52062306a36Sopenharmony_ci{
52162306a36Sopenharmony_ci	int err;
52262306a36Sopenharmony_ci
52362306a36Sopenharmony_ci	if (!of_machine_is_compatible("nvidia,tegra30"))
52462306a36Sopenharmony_ci		return 0;
52562306a36Sopenharmony_ci
52662306a36Sopenharmony_ci	err = register_reboot_notifier(&tegra30_coupler.reboot_notifier);
52762306a36Sopenharmony_ci	WARN_ON(err);
52862306a36Sopenharmony_ci
52962306a36Sopenharmony_ci	err = register_pm_notifier(&tegra30_coupler.suspend_notifier);
53062306a36Sopenharmony_ci	WARN_ON(err);
53162306a36Sopenharmony_ci
53262306a36Sopenharmony_ci	return regulator_coupler_register(&tegra30_coupler.coupler);
53362306a36Sopenharmony_ci}
53462306a36Sopenharmony_ciarch_initcall(tegra_regulator_coupler_init);
535