On Tue, Sep 02, 2025 at 03:46:52PM -0700, Ankur Arora wrote: > Catalin Marinas <catalin.marinas@xxxxxxx> writes: > > Can you have a go at poll_idle() to see how it would look like using > > this API? It doesn't necessarily mean we have to merge them all at once > > but it gives us a better idea of the suitability of the interface. > > So, I've been testing with some version of the following: > > diff --git a/drivers/cpuidle/poll_state.c b/drivers/cpuidle/poll_state.c > index 9b6d90a72601..361879396d0c 100644 > --- a/drivers/cpuidle/poll_state.c > +++ b/drivers/cpuidle/poll_state.c > @@ -8,35 +8,25 @@ > #include <linux/sched/clock.h> > #include <linux/sched/idle.h> > > -#define POLL_IDLE_RELAX_COUNT 200 > - > static int __cpuidle poll_idle(struct cpuidle_device *dev, > struct cpuidle_driver *drv, int index) > { > - u64 time_start; > - > - time_start = local_clock_noinstr(); > + unsigned long flags; > > dev->poll_time_limit = false; > > raw_local_irq_enable(); > if (!current_set_polling_and_test()) { > - unsigned int loop_count = 0; > - u64 limit; > + u64 limit, time_end; > > limit = cpuidle_poll_time(drv, dev); > + time_end = local_clock_noinstr() + limit; > > - while (!need_resched()) { > - cpu_relax(); > - if (loop_count++ < POLL_IDLE_RELAX_COUNT) > - continue; > + flags = smp_cond_load_relaxed_timewait(¤t_thread_info()->flags, > + VAL & _TIF_NEED_RESCHED, > + (local_clock_noinstr() >= time_end)); It makes sense to have the non-strict comparison, though it changes the original behaviour slightly. Just mention it in the commit log. > > - loop_count = 0; > - if (local_clock_noinstr() - time_start > limit) { > - dev->poll_time_limit = true; > - break; > - } > - } > + dev->poll_time_limit = (local_clock_noinstr() >= time_end); Could we do this instead and avoid another clock read: dev->poll_time_limit = !(flags & _TIF_NEED_RESCHED); In the original code, it made sense since it had to check the clock anyway and break the loop. When you repost, please include the rqspinlock and poll_idle changes as well to show how the interface is used. -- Catalin