[PATCH v2 0/9] s390: Improve this_cpu operations

Heiko Carstens posted 9 patches 2 weeks, 3 days ago
arch/s390/boot/alternative.c         |   7 +
arch/s390/include/asm/alternative.h  |   5 +
arch/s390/include/asm/entry-percpu.h |  57 ++++++
arch/s390/include/asm/lowcore.h      |   3 +-
arch/s390/include/asm/percpu.h       | 259 ++++++++++++++++++++++-----
arch/s390/include/asm/ptrace.h       |   2 +
arch/s390/kernel/alternative.c       |  25 ++-
arch/s390/kernel/irq.c               |  17 +-
arch/s390/kernel/nmi.c               |   3 +
arch/s390/kernel/traps.c             |   3 +
10 files changed, 330 insertions(+), 51 deletions(-)
create mode 100644 arch/s390/include/asm/entry-percpu.h
[PATCH v2 0/9] s390: Improve this_cpu operations
Posted by Heiko Carstens 2 weeks, 3 days ago
v2:

- Add proper PERCPU_PTR cast to most patches to avoid tons of sparse
  warnings

- Add missing __packed attribute to insn structure [Sashiko [2]]

- Fix inverted if condition [Sashiko [2]]

- Add missing user_mode() check [Sashiko [2]]

- Move percpu_entry() call in front of irqentry_enter() call in all
  entry paths to avoid that potential this_cpu() operations overwrite
  the not-yet saved percpu code section indicator  [Sashiko [2]]

[2] https://sashiko.dev/#/patchset/20260317195436.2276810-1-hca%40linux.ibm.com

v1:

This is a follow-up to Peter Zijlstra's in-kernel rseq RFC [1].

With the intended removal of PREEMPT_NONE this_cpu operations based on
atomic instructions, guarded with preempt_disable()/preempt_enable() pairs,
become more expensive: the preempt_disable() / preempt_enable() pairs are
not optimized away anymore during compile time.

In particular the conditional call to preempt_schedule_notrace() after
preempt_enable() adds additional code and register pressure.

To avoid this Peter suggested an in-kernel rseq approach. While this would
certainly work, this series tries to come up with a solution which uses
less instructions and doesn't require to repeat instruction sequences.

The idea is that this_cpu operations based on atomic instructions are
guarded with mvyi instructions:

- The first mvyi instruction writes the register number, which contains
  the percpu address variable to lowcore. This also indicates that a
  percpu code section is executed.

- The first instruction following the mvyi instruction must be the ag
  instruction which adds the percpu offset to the percpu address register.

- Afterwards the atomic percpu operation follows.

- Then a second mvyi instruction writes a zero to lowcore, which indicates
  the end of the percpu code section.

- In case of an interrupt/exception/nmi the register number which was
  written to lowcore is copied to the exception frame (pt_regs), and a zero
  is written to lowcore.

- On return to the previous context it is checked if a percpu code section
  was executed (saved register number not zero), and if the process was
  migrated to a different cpu. If the percpu offset was already added to
  the percpu address register (instruction address does _not_ point to the
  ag instruction) the content of the percpu address register is adjusted so
  it points to percpu variable of the new cpu.

All of this seems to work, but of course it could still be broken since I
missed some detail.

In total this series results in a kernel text size reduction of ~106kb. The
number of preempt_schedule_notrace() call sites is reduced from 7089 to
1577.

Note: this comes without any huge performance analysis, however all
microbenchmarks confirmed that the new code is at least as fast as the
old code, like expected.

[1] 20260223163843.GR1282955@noisy.programming.kicks-ass.net

Heiko Carstens (9):
  s390/percpu: Provide arch_raw_cpu_ptr()
  s390/alternatives: Add new ALT_TYPE_PERCPU type
  s390/percpu: Infrastructure for more efficient this_cpu operations
  s390/percpu: Use new percpu code section for arch_this_cpu_add()
  s390/percpu: Use new percpu code section for arch_this_cpu_add_return()
  s390/percpu: Use new percpu code section for arch_this_cpu_[and|or]()
  s390/percpu: Provide arch_this_cpu_read() implementation
  s390/percpu: Provide arch_this_cpu_write() implementation
  s390/percpu: Remove one and two byte this_cpu operation implementation

 arch/s390/boot/alternative.c         |   7 +
 arch/s390/include/asm/alternative.h  |   5 +
 arch/s390/include/asm/entry-percpu.h |  57 ++++++
 arch/s390/include/asm/lowcore.h      |   3 +-
 arch/s390/include/asm/percpu.h       | 259 ++++++++++++++++++++++-----
 arch/s390/include/asm/ptrace.h       |   2 +
 arch/s390/kernel/alternative.c       |  25 ++-
 arch/s390/kernel/irq.c               |  17 +-
 arch/s390/kernel/nmi.c               |   3 +
 arch/s390/kernel/traps.c             |   3 +
 10 files changed, 330 insertions(+), 51 deletions(-)
 create mode 100644 arch/s390/include/asm/entry-percpu.h

-- 
2.51.0
Re: [PATCH v2 0/9] s390: Improve this_cpu operations
Posted by Peter Zijlstra 2 weeks, 3 days ago
On Thu, Mar 19, 2026 at 01:04:54PM +0100, Heiko Carstens wrote:
> v2:
> 
> - Add proper PERCPU_PTR cast to most patches to avoid tons of sparse
>   warnings
> 
> - Add missing __packed attribute to insn structure [Sashiko [2]]
> 
> - Fix inverted if condition [Sashiko [2]]
> 
> - Add missing user_mode() check [Sashiko [2]]
> 
> - Move percpu_entry() call in front of irqentry_enter() call in all
>   entry paths to avoid that potential this_cpu() operations overwrite
>   the not-yet saved percpu code section indicator  [Sashiko [2]]

Would it make sense to add arch hooks to irqentry_{enter,exit}() ?
Re: [PATCH v2 0/9] s390: Improve this_cpu operations
Posted by Heiko Carstens 2 weeks, 2 days ago
On Thu, Mar 19, 2026 at 02:56:12PM +0100, Peter Zijlstra wrote:
> On Thu, Mar 19, 2026 at 01:04:54PM +0100, Heiko Carstens wrote:
> > v2:
> > 
> > - Add proper PERCPU_PTR cast to most patches to avoid tons of sparse
> >   warnings
> > 
> > - Add missing __packed attribute to insn structure [Sashiko [2]]
> > 
> > - Fix inverted if condition [Sashiko [2]]
> > 
> > - Add missing user_mode() check [Sashiko [2]]
> > 
> > - Move percpu_entry() call in front of irqentry_enter() call in all
> >   entry paths to avoid that potential this_cpu() operations overwrite
> >   the not-yet saved percpu code section indicator  [Sashiko [2]]
> 
> Would it make sense to add arch hooks to irqentry_{enter,exit}() ?

I guess it would make sense to have some architecture hook which
allows to run code on all entry/exit paths instead of duplicating the
code n times.

But apparently my code seems to have more bugs (e.g. I didn't consider
kprobes which would make the instruction comparison not work). So this
has to wait until I'm back after vacation.
Re: [PATCH v2 0/9] s390: Improve this_cpu operations
Posted by Peter Zijlstra 2 weeks, 2 days ago
On Fri, Mar 20, 2026 at 12:39:09PM +0100, Heiko Carstens wrote:
> On Thu, Mar 19, 2026 at 02:56:12PM +0100, Peter Zijlstra wrote:
> > On Thu, Mar 19, 2026 at 01:04:54PM +0100, Heiko Carstens wrote:
> > > v2:
> > > 
> > > - Add proper PERCPU_PTR cast to most patches to avoid tons of sparse
> > >   warnings
> > > 
> > > - Add missing __packed attribute to insn structure [Sashiko [2]]
> > > 
> > > - Fix inverted if condition [Sashiko [2]]
> > > 
> > > - Add missing user_mode() check [Sashiko [2]]
> > > 
> > > - Move percpu_entry() call in front of irqentry_enter() call in all
> > >   entry paths to avoid that potential this_cpu() operations overwrite
> > >   the not-yet saved percpu code section indicator  [Sashiko [2]]
> > 
> > Would it make sense to add arch hooks to irqentry_{enter,exit}() ?
> 
> I guess it would make sense to have some architecture hook which
> allows to run code on all entry/exit paths instead of duplicating the
> code n times.
> 
> But apparently my code seems to have more bugs (e.g. I didn't consider
> kprobes which would make the instruction comparison not work). So this
> has to wait until I'm back after vacation.

My vote is to reject kprobes in these regions :-) Anyway, enjoy the time
off!