cpu_pm: Switch to mainline implementation