writel(ctrl, gt_base + GT_CONTROL);
 }
 
-static void gt_clockevent_set_mode(enum clock_event_mode mode,
-                                  struct clock_event_device *clk)
+static int gt_clockevent_shutdown(struct clock_event_device *evt)
 {
        unsigned long ctrl;
 
-       switch (mode) {
-       case CLOCK_EVT_MODE_PERIODIC:
-               gt_compare_set(DIV_ROUND_CLOSEST(gt_clk_rate, HZ), 1);
-               break;
-       case CLOCK_EVT_MODE_ONESHOT:
-       case CLOCK_EVT_MODE_UNUSED:
-       case CLOCK_EVT_MODE_SHUTDOWN:
-               ctrl = readl(gt_base + GT_CONTROL);
-               ctrl &= ~(GT_CONTROL_COMP_ENABLE |
-                               GT_CONTROL_IRQ_ENABLE | GT_CONTROL_AUTO_INC);
-               writel(ctrl, gt_base + GT_CONTROL);
-               break;
-       default:
-               break;
-       }
+       ctrl = readl(gt_base + GT_CONTROL);
+       ctrl &= ~(GT_CONTROL_COMP_ENABLE | GT_CONTROL_IRQ_ENABLE |
+                 GT_CONTROL_AUTO_INC);
+       writel(ctrl, gt_base + GT_CONTROL);
+       return 0;
+}
+
+static int gt_clockevent_set_periodic(struct clock_event_device *evt)
+{
+       gt_compare_set(DIV_ROUND_CLOSEST(gt_clk_rate, HZ), 1);
+       return 0;
 }
 
 static int gt_clockevent_set_next_event(unsigned long evt,
         *      the Global Timer flag _after_ having incremented
         *      the Comparator register value to a higher value.
         */
-       if (evt->mode == CLOCK_EVT_MODE_ONESHOT)
+       if (clockevent_state_oneshot(evt))
                gt_compare_set(ULONG_MAX, 0);
 
        writel_relaxed(GT_INT_STATUS_EVENT_FLAG, gt_base + GT_INT_STATUS);
        clk->name = "arm_global_timer";
        clk->features = CLOCK_EVT_FEAT_PERIODIC | CLOCK_EVT_FEAT_ONESHOT |
                CLOCK_EVT_FEAT_PERCPU;
-       clk->set_mode = gt_clockevent_set_mode;
+       clk->set_state_shutdown = gt_clockevent_shutdown;
+       clk->set_state_periodic = gt_clockevent_set_periodic;
+       clk->set_state_oneshot = gt_clockevent_shutdown;
        clk->set_next_event = gt_clockevent_set_next_event;
        clk->cpumask = cpumask_of(cpu);
        clk->rating = 300;
 
 static void gt_clockevents_stop(struct clock_event_device *clk)
 {
-       gt_clockevent_set_mode(CLOCK_EVT_MODE_UNUSED, clk);
+       gt_clockevent_shutdown(clk);
        disable_percpu_irq(clk->irq);
 }