/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __UM_DELAY_H
#define __UM_DELAY_H
#include <asm-generic/delay.h>
#include <linux/time-internal.h>

static inline void um_ndelay(unsigned long nsecs)
{
 if (time_travel_mode == TT_MODE_INFCPU ||
     time_travel_mode == TT_MODE_EXTERNAL) {
  time_travel_ndelay(nsecs);
  return;
 }
 ndelay(nsecs);
}
#undef ndelay
#define ndelay(n) um_ndelay(n)

static inline void um_udelay(unsigned long usecs)
{
 if (time_travel_mode == TT_MODE_INFCPU ||
     time_travel_mode == TT_MODE_EXTERNAL) {
  time_travel_ndelay(1000 * usecs);
  return;
 }
 udelay(usecs);
}
#undef udelay
#define udelay(n) um_udelay(n)
#endif /* __UM_DELAY_H */