1 files changed, 10 insertions, 29 deletions

diff --git a/drivers/clocksource/arm_global_timer.c b/drivers/clocksource/arm_global_timer.c
index 2a9ceb6e93f9..8da03298f844 100644
--- a/drivers/clocksource/arm_global_timer.c
+++ b/drivers/clocksource/arm_global_timer.c
@@ -165,9 +165,9 @@ static irqreturn_t gt_clockevent_interrupt(int irq, void *dev_id)
         return IRQ_HANDLED;
 }
 
-static int gt_clockevents_init(struct clock_event_device *clk)
+static int gt_starting_cpu(unsigned int cpu)
 {
-        int cpu = smp_processor_id();
+        struct clock_event_device *clk = this_cpu_ptr(gt_evt);
 
         clk->name = "arm_global_timer";
         clk->features = CLOCK_EVT_FEAT_PERIODIC | CLOCK_EVT_FEAT_ONESHOT |
@@ -186,10 +186,13 @@ static int gt_clockevents_init(struct clock_event_device *clk)
         return 0;
 }
 
-static void gt_clockevents_stop(struct clock_event_device *clk)
+static int gt_dying_cpu(unsigned int cpu)
 {
+        struct clock_event_device *clk = this_cpu_ptr(gt_evt);
+
         gt_clockevent_shutdown(clk);
         disable_percpu_irq(clk->irq);
+        return 0;
 }
 
 static cycle_t gt_clocksource_read(struct clocksource *cs)
@@ -252,24 +255,6 @@ static int __init gt_clocksource_init(void)
         return clocksource_register_hz(&gt_clocksource, gt_clk_rate);
 }
 
-static int gt_cpu_notify(struct notifier_block *self, unsigned long action,
-                         void *hcpu)
-{
-        switch (action & ~CPU_TASKS_FROZEN) {
-        case CPU_STARTING:
-                gt_clockevents_init(this_cpu_ptr(gt_evt));
-                break;
-        case CPU_DYING:
-                gt_clockevents_stop(this_cpu_ptr(gt_evt));
-                break;
-        }
-
-        return NOTIFY_OK;
-}
-static struct notifier_block gt_cpu_nb = {
-        .notifier_call = gt_cpu_notify,
-};
-
 static int __init global_timer_of_register(struct device_node *np)
 {
         struct clk *gt_clk;
@@ -325,18 +310,14 @@ static int __init global_timer_of_register(struct device_node *np)
                 goto out_free;
         }
 
-        err = register_cpu_notifier(&gt_cpu_nb);
-        if (err) {
-                pr_warn("global-timer: unable to register cpu notifier.\n");
-                goto out_irq;
-        }
-
-        /* Immediately configure the timer on the boot CPU */
+        /* Register and immediately configure the timer on the boot CPU */
         err = gt_clocksource_init();
         if (err)
                 goto out_irq;
         
-        err = gt_clockevents_init(this_cpu_ptr(gt_evt));
+        err = cpuhp_setup_state(CPUHP_AP_ARM_GLOBAL_TIMER_STARTING,
+                                "AP_ARM_GLOBAL_TIMER_STARTING",
+                                gt_starting_cpu, gt_dying_cpu);
         if (err)
                 goto out_irq;