if (uts) {
                struct timespec64 ts;
+               ktime_t dt;
 
                if (get_timespec64(&ts, uts))
                        return -EFAULT;
 
-               iowq.timeout = ktime_add_ns(timespec64_to_ktime(ts), ktime_get_ns());
-               io_napi_adjust_timeout(ctx, &iowq, &ts);
+               dt = timespec64_to_ktime(ts);
+               iowq.timeout = ktime_add(dt, ktime_get());
+               io_napi_adjust_timeout(ctx, &iowq, dt);
        }
 
        if (sig) {
 
  * the NAPI timeout accordingly.
  */
 void __io_napi_adjust_timeout(struct io_ring_ctx *ctx, struct io_wait_queue *iowq,
-                             struct timespec64 *ts)
+                             ktime_t to_wait)
 {
        ktime_t poll_dt = READ_ONCE(ctx->napi_busy_poll_dt);
 
-       if (ts) {
-               struct timespec64 poll_to_ts;
-
-               poll_to_ts = ns_to_timespec64(ktime_to_ns(poll_dt));
-               if (timespec64_compare(ts, &poll_to_ts) < 0) {
-                       s64 poll_to_ns = timespec64_to_ns(ts);
-                       if (poll_to_ns > 0)
-                               poll_dt = ns_to_ktime(poll_to_ns);
-               }
-       }
+       if (to_wait)
+               poll_dt = min(poll_dt, to_wait);
 
        iowq->napi_busy_poll_dt = poll_dt;
 }
 
 void __io_napi_add(struct io_ring_ctx *ctx, struct socket *sock);
 
 void __io_napi_adjust_timeout(struct io_ring_ctx *ctx,
-               struct io_wait_queue *iowq, struct timespec64 *ts);
+               struct io_wait_queue *iowq, ktime_t to_wait);
 void __io_napi_busy_loop(struct io_ring_ctx *ctx, struct io_wait_queue *iowq);
 int io_napi_sqpoll_busy_poll(struct io_ring_ctx *ctx);
 
 
 static inline void io_napi_adjust_timeout(struct io_ring_ctx *ctx,
                                          struct io_wait_queue *iowq,
-                                         struct timespec64 *ts)
+                                         ktime_t to_wait)
 {
        if (!io_napi(ctx))
                return;
-       __io_napi_adjust_timeout(ctx, iowq, ts);
+       __io_napi_adjust_timeout(ctx, iowq, to_wait);
 }
 
 static inline void io_napi_busy_loop(struct io_ring_ctx *ctx,
 }
 static inline void io_napi_adjust_timeout(struct io_ring_ctx *ctx,
                                          struct io_wait_queue *iowq,
-                                         struct timespec64 *ts)
+                                         ktime_t to_wait)
 {
 }
 static inline void io_napi_busy_loop(struct io_ring_ctx *ctx,