1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * Copyright (C) 2016 Freescale Semiconductor, Inc.
4 * Copyright 2017-2018 NXP
5 *   Anson Huang <Anson.Huang@nxp.com>
6 */
7
8#include <linux/cpuidle.h>
9#include <linux/module.h>
10#include <asm/cpuidle.h>
11
12#include "common.h"
13#include "cpuidle.h"
14
15static __cpuidle int imx7ulp_enter_wait(struct cpuidle_device *dev,
16					struct cpuidle_driver *drv, int index)
17{
18	if (index == 1)
19		imx7ulp_set_lpm(ULP_PM_WAIT);
20	else
21		imx7ulp_set_lpm(ULP_PM_STOP);
22
23	cpu_do_idle();
24
25	imx7ulp_set_lpm(ULP_PM_RUN);
26
27	return index;
28}
29
30static struct cpuidle_driver imx7ulp_cpuidle_driver = {
31	.name = "imx7ulp_cpuidle",
32	.owner = THIS_MODULE,
33	.states = {
34		/* WFI */
35		ARM_CPUIDLE_WFI_STATE,
36		/* WAIT */
37		{
38			.exit_latency = 50,
39			.target_residency = 75,
40			.enter = imx7ulp_enter_wait,
41			.name = "WAIT",
42			.desc = "PSTOP2",
43		},
44		/* STOP */
45		{
46			.exit_latency = 100,
47			.target_residency = 150,
48			.enter = imx7ulp_enter_wait,
49			.name = "STOP",
50			.desc = "PSTOP1",
51		},
52	},
53	.state_count = 3,
54	.safe_state_index = 0,
55};
56
57int __init imx7ulp_cpuidle_init(void)
58{
59	return cpuidle_register(&imx7ulp_cpuidle_driver, NULL);
60}
61