@@ -21,6 +21,7 @@
#include <linux/cpu_pm.h>
#include <linux/kernel.h>
#include <linux/init.h>
+#include <linux/prctl.h>
#include <linux/sched/signal.h>
#include <linux/signal.h>
#include <linux/hardirq.h>
@@ -226,8 +227,17 @@ int sve_set_vector_length(struct task_struct *task,
{
BUG_ON(task == current && preemptible());
+ /*
+ * To avoid accidents, forbid setting for individual threads of a
+ * multithreaded process. User code that knows what it's doing can
+ * pass PR_SVE_SET_VL_THREAD to override this restriction:
+ */
+ if (!(flags & PR_SVE_SET_VL_THREAD) && get_nr_threads(task) != 1)
+ return -EINVAL;
+
+ flags &= ~(unsigned long)PR_SVE_SET_VL_THREAD;
if (flags)
- return -EINVAL; /* No flags defined yet */
+ return -EINVAL; /* No other flags defined yet */
if (!sve_vl_valid(vl))
return -EINVAL;
@@ -199,6 +199,7 @@ struct prctl_mm_map {
/* arm64 Scalable Vector Extension controls */
#define PR_SVE_SET_VL 48 /* set task vector length */
+# define PR_SVE_SET_VL_THREAD (1 << 1) /* set just this thread */
#define PR_SVE_GET_VL 49 /* get task vector length */
#endif /* _LINUX_PRCTL_H */
General-purpose code in userspace is not expected to work correctly if multiple threads are allowed to run concurrently with different vector lengths in a single process. This patch adds an explicit flag PR_SVE_SET_VL_THREAD to request this behaviour. Without the flag, vector length setting is permitted only for a single-threaded process (which matches the expected usage model of setting the vector length at process startup). Signed-off-by: Dave Martin <Dave.Martin@arm.com> --- arch/arm64/kernel/fpsimd.c | 12 +++++++++++- include/uapi/linux/prctl.h | 1 + 2 files changed, 12 insertions(+), 1 deletion(-)