|  | // SPDX-License-Identifier: GPL-2.0+ | 
|  | /* | 
|  | * Copyright (C) 2016 Freescale Semiconductor, Inc. | 
|  | * Copyright 2017-2018 NXP | 
|  | *   Anson Huang <Anson.Huang@nxp.com> | 
|  | */ | 
|  |  | 
|  | #include <linux/cpuidle.h> | 
|  | #include <linux/module.h> | 
|  | #include <asm/cpuidle.h> | 
|  |  | 
|  | #include "common.h" | 
|  | #include "cpuidle.h" | 
|  |  | 
|  | static __cpuidle int imx7ulp_enter_wait(struct cpuidle_device *dev, | 
|  | struct cpuidle_driver *drv, int index) | 
|  | { | 
|  | if (index == 1) | 
|  | imx7ulp_set_lpm(ULP_PM_WAIT); | 
|  | else | 
|  | imx7ulp_set_lpm(ULP_PM_STOP); | 
|  |  | 
|  | cpu_do_idle(); | 
|  |  | 
|  | imx7ulp_set_lpm(ULP_PM_RUN); | 
|  |  | 
|  | return index; | 
|  | } | 
|  |  | 
|  | static struct cpuidle_driver imx7ulp_cpuidle_driver = { | 
|  | .name = "imx7ulp_cpuidle", | 
|  | .owner = THIS_MODULE, | 
|  | .states = { | 
|  | /* WFI */ | 
|  | ARM_CPUIDLE_WFI_STATE, | 
|  | /* WAIT */ | 
|  | { | 
|  | .exit_latency = 50, | 
|  | .target_residency = 75, | 
|  | .enter = imx7ulp_enter_wait, | 
|  | .name = "WAIT", | 
|  | .desc = "PSTOP2", | 
|  | }, | 
|  | /* STOP */ | 
|  | { | 
|  | .exit_latency = 100, | 
|  | .target_residency = 150, | 
|  | .enter = imx7ulp_enter_wait, | 
|  | .name = "STOP", | 
|  | .desc = "PSTOP1", | 
|  | }, | 
|  | }, | 
|  | .state_count = 3, | 
|  | .safe_state_index = 0, | 
|  | }; | 
|  |  | 
|  | int __init imx7ulp_cpuidle_init(void) | 
|  | { | 
|  | return cpuidle_register(&imx7ulp_cpuidle_driver, NULL); | 
|  | } |