enable_percpu_irq(clk->irq, 0);
 }
 
-static int twd_timer_cpu_notify(struct notifier_block *self,
-                               unsigned long action, void *hcpu)
+static int twd_timer_starting_cpu(unsigned int cpu)
 {
-       switch (action & ~CPU_TASKS_FROZEN) {
-       case CPU_STARTING:
-               twd_timer_setup();
-               break;
-       case CPU_DYING:
-               twd_timer_stop();
-               break;
-       }
-
-       return NOTIFY_OK;
+       twd_timer_setup();
+       return 0;
 }
 
-static struct notifier_block twd_timer_cpu_nb = {
-       .notifier_call = twd_timer_cpu_notify,
-};
+static int twd_timer_dying_cpu(unsigned int cpu)
+{
+       twd_timer_stop();
+       return 0;
+}
 
 static int __init twd_local_timer_common_register(struct device_node *np)
 {
                goto out_free;
        }
 
-       err = register_cpu_notifier(&twd_timer_cpu_nb);
-       if (err)
-               goto out_irq;
+       cpuhp_setup_state_nocalls(CPUHP_AP_ARM_TWD_STARTING,
+                                 "AP_ARM_TWD_STARTING",
+                                 twd_timer_starting_cpu, twd_timer_dying_cpu);
 
        twd_get_clock(np);
        if (!of_property_read_bool(np, "always-on"))
 
        return 0;
 
-out_irq:
-       free_percpu_irq(twd_ppi, twd_evt);
 out_free:
        iounmap(twd_base);
        twd_base = NULL;
 
        CPUHP_AP_ARM_L2X0_STARTING,
        CPUHP_AP_ARM_ARCH_TIMER_STARTING,
        CPUHP_AP_DUMMY_TIMER_STARTING,
+       CPUHP_AP_ARM_TWD_STARTING,
        CPUHP_AP_METAG_TIMER_STARTING,
        CPUHP_AP_QCOM_TIMER_STARTING,
        CPUHP_AP_MIPS_GIC_TIMER_STARTING,