return 0;
 }
 
-static inline void rk_timer_set_mode(enum clock_event_mode mode,
-                                    struct clock_event_device *ce)
+static int rk_timer_shutdown(struct clock_event_device *ce)
 {
-       switch (mode) {
-       case CLOCK_EVT_MODE_PERIODIC:
-               rk_timer_disable(ce);
-               rk_timer_update_counter(rk_timer(ce)->freq / HZ - 1, ce);
-               rk_timer_enable(ce, TIMER_MODE_FREE_RUNNING);
-               break;
-       case CLOCK_EVT_MODE_ONESHOT:
-       case CLOCK_EVT_MODE_RESUME:
-               break;
-       case CLOCK_EVT_MODE_UNUSED:
-       case CLOCK_EVT_MODE_SHUTDOWN:
-               rk_timer_disable(ce);
-               break;
-       }
+       rk_timer_disable(ce);
+       return 0;
+}
+
+static int rk_timer_set_periodic(struct clock_event_device *ce)
+{
+       rk_timer_disable(ce);
+       rk_timer_update_counter(rk_timer(ce)->freq / HZ - 1, ce);
+       rk_timer_enable(ce, TIMER_MODE_FREE_RUNNING);
+       return 0;
 }
 
 static irqreturn_t rk_timer_interrupt(int irq, void *dev_id)
 
        rk_timer_interrupt_clear(ce);
 
-       if (ce->mode == CLOCK_EVT_MODE_ONESHOT)
+       if (clockevent_state_oneshot(ce))
                rk_timer_disable(ce);
 
        ce->event_handler(ce);
        ce->name = TIMER_NAME;
        ce->features = CLOCK_EVT_FEAT_PERIODIC | CLOCK_EVT_FEAT_ONESHOT;
        ce->set_next_event = rk_timer_set_next_event;
-       ce->set_mode = rk_timer_set_mode;
+       ce->set_state_shutdown = rk_timer_shutdown;
+       ce->set_state_periodic = rk_timer_set_periodic;
        ce->irq = irq;
        ce->cpumask = cpumask_of(0);
        ce->rating = 250;