1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3 * Copyright (C) 2020 SiFive
4 */
5
6 #ifndef __ASM_RISCV_VECTOR_H
7 #define __ASM_RISCV_VECTOR_H
8
9 #include <linux/types.h>
10 #include <uapi/asm-generic/errno.h>
11
12 #ifdef CONFIG_RISCV_ISA_V
13
14 #include <linux/stringify.h>
15 #include <linux/sched.h>
16 #include <linux/sched/task_stack.h>
17 #include <asm/ptrace.h>
18 #include <asm/hwcap.h>
19 #include <asm/csr.h>
20 #include <asm/asm.h>
21
22 extern unsigned long riscv_v_vsize;
23 int riscv_v_setup_vsize(void);
24 bool riscv_v_first_use_handler(struct pt_regs *regs);
25
has_vector(void)26 static __always_inline bool has_vector(void)
27 {
28 return riscv_has_extension_unlikely(RISCV_ISA_EXT_v);
29 }
30
__riscv_v_vstate_clean(struct pt_regs * regs)31 static inline void __riscv_v_vstate_clean(struct pt_regs *regs)
32 {
33 regs->status = (regs->status & ~SR_VS) | SR_VS_CLEAN;
34 }
35
__riscv_v_vstate_dirty(struct pt_regs * regs)36 static inline void __riscv_v_vstate_dirty(struct pt_regs *regs)
37 {
38 regs->status = (regs->status & ~SR_VS) | SR_VS_DIRTY;
39 }
40
riscv_v_vstate_off(struct pt_regs * regs)41 static inline void riscv_v_vstate_off(struct pt_regs *regs)
42 {
43 regs->status = (regs->status & ~SR_VS) | SR_VS_OFF;
44 }
45
riscv_v_vstate_on(struct pt_regs * regs)46 static inline void riscv_v_vstate_on(struct pt_regs *regs)
47 {
48 regs->status = (regs->status & ~SR_VS) | SR_VS_INITIAL;
49 }
50
riscv_v_vstate_query(struct pt_regs * regs)51 static inline bool riscv_v_vstate_query(struct pt_regs *regs)
52 {
53 return (regs->status & SR_VS) != 0;
54 }
55
riscv_v_enable(void)56 static __always_inline void riscv_v_enable(void)
57 {
58 csr_set(CSR_SSTATUS, SR_VS);
59 }
60
riscv_v_disable(void)61 static __always_inline void riscv_v_disable(void)
62 {
63 csr_clear(CSR_SSTATUS, SR_VS);
64 }
65
__vstate_csr_save(struct __riscv_v_ext_state * dest)66 static __always_inline void __vstate_csr_save(struct __riscv_v_ext_state *dest)
67 {
68 asm volatile (
69 "csrr %0, " __stringify(CSR_VSTART) "\n\t"
70 "csrr %1, " __stringify(CSR_VTYPE) "\n\t"
71 "csrr %2, " __stringify(CSR_VL) "\n\t"
72 "csrr %3, " __stringify(CSR_VCSR) "\n\t"
73 "csrr %4, " __stringify(CSR_VLENB) "\n\t"
74 : "=r" (dest->vstart), "=r" (dest->vtype), "=r" (dest->vl),
75 "=r" (dest->vcsr), "=r" (dest->vlenb) : :);
76 }
77
__vstate_csr_restore(struct __riscv_v_ext_state * src)78 static __always_inline void __vstate_csr_restore(struct __riscv_v_ext_state *src)
79 {
80 asm volatile (
81 ".option push\n\t"
82 ".option arch, +v\n\t"
83 "vsetvl x0, %2, %1\n\t"
84 ".option pop\n\t"
85 "csrw " __stringify(CSR_VSTART) ", %0\n\t"
86 "csrw " __stringify(CSR_VCSR) ", %3\n\t"
87 : : "r" (src->vstart), "r" (src->vtype), "r" (src->vl),
88 "r" (src->vcsr) :);
89 }
90
__riscv_v_vstate_save(struct __riscv_v_ext_state * save_to,void * datap)91 static inline void __riscv_v_vstate_save(struct __riscv_v_ext_state *save_to,
92 void *datap)
93 {
94 unsigned long vl;
95
96 riscv_v_enable();
97 __vstate_csr_save(save_to);
98 asm volatile (
99 ".option push\n\t"
100 ".option arch, +v\n\t"
101 "vsetvli %0, x0, e8, m8, ta, ma\n\t"
102 "vse8.v v0, (%1)\n\t"
103 "add %1, %1, %0\n\t"
104 "vse8.v v8, (%1)\n\t"
105 "add %1, %1, %0\n\t"
106 "vse8.v v16, (%1)\n\t"
107 "add %1, %1, %0\n\t"
108 "vse8.v v24, (%1)\n\t"
109 ".option pop\n\t"
110 : "=&r" (vl) : "r" (datap) : "memory");
111 riscv_v_disable();
112 }
113
__riscv_v_vstate_restore(struct __riscv_v_ext_state * restore_from,void * datap)114 static inline void __riscv_v_vstate_restore(struct __riscv_v_ext_state *restore_from,
115 void *datap)
116 {
117 unsigned long vl;
118
119 riscv_v_enable();
120 asm volatile (
121 ".option push\n\t"
122 ".option arch, +v\n\t"
123 "vsetvli %0, x0, e8, m8, ta, ma\n\t"
124 "vle8.v v0, (%1)\n\t"
125 "add %1, %1, %0\n\t"
126 "vle8.v v8, (%1)\n\t"
127 "add %1, %1, %0\n\t"
128 "vle8.v v16, (%1)\n\t"
129 "add %1, %1, %0\n\t"
130 "vle8.v v24, (%1)\n\t"
131 ".option pop\n\t"
132 : "=&r" (vl) : "r" (datap) : "memory");
133 __vstate_csr_restore(restore_from);
134 riscv_v_disable();
135 }
136
__riscv_v_vstate_discard(void)137 static inline void __riscv_v_vstate_discard(void)
138 {
139 unsigned long vl, vtype_inval = 1UL << (BITS_PER_LONG - 1);
140
141 riscv_v_enable();
142 asm volatile (
143 ".option push\n\t"
144 ".option arch, +v\n\t"
145 "vsetvli %0, x0, e8, m8, ta, ma\n\t"
146 "vmv.v.i v0, -1\n\t"
147 "vmv.v.i v8, -1\n\t"
148 "vmv.v.i v16, -1\n\t"
149 "vmv.v.i v24, -1\n\t"
150 "vsetvl %0, x0, %1\n\t"
151 ".option pop\n\t"
152 : "=&r" (vl) : "r" (vtype_inval) : "memory");
153 riscv_v_disable();
154 }
155
riscv_v_vstate_discard(struct pt_regs * regs)156 static inline void riscv_v_vstate_discard(struct pt_regs *regs)
157 {
158 if ((regs->status & SR_VS) == SR_VS_OFF)
159 return;
160
161 __riscv_v_vstate_discard();
162 __riscv_v_vstate_dirty(regs);
163 }
164
riscv_v_vstate_save(struct task_struct * task,struct pt_regs * regs)165 static inline void riscv_v_vstate_save(struct task_struct *task,
166 struct pt_regs *regs)
167 {
168 if ((regs->status & SR_VS) == SR_VS_DIRTY) {
169 struct __riscv_v_ext_state *vstate = &task->thread.vstate;
170
171 __riscv_v_vstate_save(vstate, vstate->datap);
172 __riscv_v_vstate_clean(regs);
173 }
174 }
175
riscv_v_vstate_restore(struct task_struct * task,struct pt_regs * regs)176 static inline void riscv_v_vstate_restore(struct task_struct *task,
177 struct pt_regs *regs)
178 {
179 if ((regs->status & SR_VS) != SR_VS_OFF) {
180 struct __riscv_v_ext_state *vstate = &task->thread.vstate;
181
182 __riscv_v_vstate_restore(vstate, vstate->datap);
183 __riscv_v_vstate_clean(regs);
184 }
185 }
186
__switch_to_vector(struct task_struct * prev,struct task_struct * next)187 static inline void __switch_to_vector(struct task_struct *prev,
188 struct task_struct *next)
189 {
190 struct pt_regs *regs;
191
192 regs = task_pt_regs(prev);
193 riscv_v_vstate_save(prev, regs);
194 riscv_v_vstate_restore(next, task_pt_regs(next));
195 }
196
197 void riscv_v_vstate_ctrl_init(struct task_struct *tsk);
198 bool riscv_v_vstate_ctrl_user_allowed(void);
199
200 #else /* ! CONFIG_RISCV_ISA_V */
201
202 struct pt_regs;
203
riscv_v_setup_vsize(void)204 static inline int riscv_v_setup_vsize(void) { return -EOPNOTSUPP; }
has_vector(void)205 static __always_inline bool has_vector(void) { return false; }
riscv_v_first_use_handler(struct pt_regs * regs)206 static inline bool riscv_v_first_use_handler(struct pt_regs *regs) { return false; }
riscv_v_vstate_query(struct pt_regs * regs)207 static inline bool riscv_v_vstate_query(struct pt_regs *regs) { return false; }
riscv_v_vstate_ctrl_user_allowed(void)208 static inline bool riscv_v_vstate_ctrl_user_allowed(void) { return false; }
209 #define riscv_v_vsize (0)
210 #define riscv_v_vstate_discard(regs) do {} while (0)
211 #define riscv_v_vstate_save(task, regs) do {} while (0)
212 #define riscv_v_vstate_restore(task, regs) do {} while (0)
213 #define __switch_to_vector(__prev, __next) do {} while (0)
214 #define riscv_v_vstate_off(regs) do {} while (0)
215 #define riscv_v_vstate_on(regs) do {} while (0)
216
217 #endif /* CONFIG_RISCV_ISA_V */
218
219 #endif /* ! __ASM_RISCV_VECTOR_H */
220