On Fri, Jul 17, 2020 at 10:59:54AM +0200, Peter Zijlstra wrote: > On Fri, Jul 17, 2020 at 12:20:43AM -0700, ira.weiny@xxxxxxxxx wrote: > > +/* > > + * Write the PKey Register Supervisor. This must be run with preemption > > + * disabled as it does not guarantee the atomicity of updating the pkrs_cache > > + * and MSR on its own. > > + */ > > +void write_pkrs(u32 pkrs_val) > > +{ > > + this_cpu_write(pkrs_cache, pkrs_val); > > + wrmsrl(MSR_IA32_PKRS, pkrs_val); > > +} > > Should we write that like: > > void write_pkrs(u32 pkr) > { > u32 *pkrs = get_cpu_ptr(pkrs_cache); > if (*pkrs != pkr) { > *pkrs = pkr; > wrmsrl(MSR_IA32_PKRS, pkr); > } > put_cpu_ptrpkrs_cache); > } > > given that we fundamentally need to serialize againt schedule() here. Yes. That seems better. That also means pks_sched_in() can be simplified to just static inline void pks_sched_in(void) { write_pkrs(current->thread.saved_pkrs); } Because of the built WRMSR avoidance. However, pkrs_cache is static so I think I need to use {get,put}_cpu_var() here don't I? Thanks! Ira