1 /* SPDX-License-Identifier: GPL-2.0-only */
2 /*
3  * Copyright 2022-2024 Rivos, Inc
4  */
5 
6 #ifndef _ASM_CPUFEATURE_H
7 #define _ASM_CPUFEATURE_H
8 
9 #include <linux/bitmap.h>
10 #include <linux/jump_label.h>
11 #include <linux/workqueue.h>
12 #include <linux/kconfig.h>
13 #include <linux/percpu-defs.h>
14 #include <linux/threads.h>
15 #include <asm/hwcap.h>
16 #include <asm/cpufeature-macros.h>
17 
18 /*
19  * These are probed via a device_initcall(), via either the SBI or directly
20  * from the corresponding CSRs.
21  */
22 struct riscv_cpuinfo {
23 	unsigned long mvendorid;
24 	unsigned long marchid;
25 	unsigned long mimpid;
26 };
27 
28 struct riscv_isainfo {
29 	DECLARE_BITMAP(isa, RISCV_ISA_EXT_MAX);
30 };
31 
32 DECLARE_PER_CPU(struct riscv_cpuinfo, riscv_cpuinfo);
33 
34 /* Per-cpu ISA extensions. */
35 extern struct riscv_isainfo hart_isa[NR_CPUS];
36 
37 extern u32 thead_vlenb_of;
38 
39 void __init riscv_user_isa_enable(void);
40 
41 #define _RISCV_ISA_EXT_DATA(_name, _id, _subset_exts, _subset_exts_size, _validate) {	\
42 	.name = #_name,									\
43 	.property = #_name,								\
44 	.id = _id,									\
45 	.subset_ext_ids = _subset_exts,							\
46 	.subset_ext_size = _subset_exts_size,						\
47 	.validate = _validate								\
48 }
49 
50 #define __RISCV_ISA_EXT_DATA(_name, _id) _RISCV_ISA_EXT_DATA(_name, _id, NULL, 0, NULL)
51 
52 #define __RISCV_ISA_EXT_DATA_VALIDATE(_name, _id, _validate) \
53 			_RISCV_ISA_EXT_DATA(_name, _id, NULL, 0, _validate)
54 
55 /* Used to declare pure "lasso" extension (Zk for instance) */
56 #define __RISCV_ISA_EXT_BUNDLE(_name, _bundled_exts) \
57 	_RISCV_ISA_EXT_DATA(_name, RISCV_ISA_EXT_INVALID, _bundled_exts, \
58 			    ARRAY_SIZE(_bundled_exts), NULL)
59 #define __RISCV_ISA_EXT_BUNDLE_VALIDATE(_name, _bundled_exts, _validate) \
60 	_RISCV_ISA_EXT_DATA(_name, RISCV_ISA_EXT_INVALID, _bundled_exts, \
61 			    ARRAY_SIZE(_bundled_exts), _validate)
62 
63 /* Used to declare extensions that are a superset of other extensions (Zvbb for instance) */
64 #define __RISCV_ISA_EXT_SUPERSET(_name, _id, _sub_exts) \
65 	_RISCV_ISA_EXT_DATA(_name, _id, _sub_exts, ARRAY_SIZE(_sub_exts), NULL)
66 #define __RISCV_ISA_EXT_SUPERSET_VALIDATE(_name, _id, _sub_exts, _validate) \
67 	_RISCV_ISA_EXT_DATA(_name, _id, _sub_exts, ARRAY_SIZE(_sub_exts), _validate)
68 
69 bool __init check_unaligned_access_emulated_all_cpus(void);
70 #if defined(CONFIG_RISCV_SCALAR_MISALIGNED)
71 void check_unaligned_access_emulated(struct work_struct *work __always_unused);
72 void unaligned_emulation_finish(void);
73 bool unaligned_ctl_available(void);
74 DECLARE_PER_CPU(long, misaligned_access_speed);
75 #else
unaligned_ctl_available(void)76 static inline bool unaligned_ctl_available(void)
77 {
78 	return false;
79 }
80 #endif
81 
82 bool __init check_vector_unaligned_access_emulated_all_cpus(void);
83 #if defined(CONFIG_RISCV_VECTOR_MISALIGNED)
84 void check_vector_unaligned_access_emulated(struct work_struct *work __always_unused);
85 DECLARE_PER_CPU(long, vector_misaligned_access);
86 #endif
87 
88 #if defined(CONFIG_RISCV_PROBE_UNALIGNED_ACCESS)
89 DECLARE_STATIC_KEY_FALSE(fast_unaligned_access_speed_key);
90 
has_fast_unaligned_accesses(void)91 static __always_inline bool has_fast_unaligned_accesses(void)
92 {
93 	return static_branch_likely(&fast_unaligned_access_speed_key);
94 }
95 #else
has_fast_unaligned_accesses(void)96 static __always_inline bool has_fast_unaligned_accesses(void)
97 {
98 	if (IS_ENABLED(CONFIG_HAVE_EFFICIENT_UNALIGNED_ACCESS))
99 		return true;
100 	else
101 		return false;
102 }
103 #endif
104 
105 unsigned long riscv_get_elf_hwcap(void);
106 
107 struct riscv_isa_ext_data {
108 	const unsigned int id;
109 	const char *name;
110 	const char *property;
111 	const unsigned int *subset_ext_ids;
112 	const unsigned int subset_ext_size;
113 	int (*validate)(const struct riscv_isa_ext_data *data, const unsigned long *isa_bitmap);
114 };
115 
116 extern const struct riscv_isa_ext_data riscv_isa_ext[];
117 extern const size_t riscv_isa_ext_count;
118 extern bool riscv_isa_fallback;
119 
120 unsigned long riscv_isa_extension_base(const unsigned long *isa_bitmap);
riscv_cpu_has_extension_likely(int cpu,const unsigned long ext)121 static __always_inline bool riscv_cpu_has_extension_likely(int cpu, const unsigned long ext)
122 {
123 	compiletime_assert(ext < RISCV_ISA_EXT_MAX, "ext must be < RISCV_ISA_EXT_MAX");
124 
125 	if (IS_ENABLED(CONFIG_RISCV_ALTERNATIVE) &&
126 	    __riscv_has_extension_likely(STANDARD_EXT, ext))
127 		return true;
128 
129 	return __riscv_isa_extension_available(hart_isa[cpu].isa, ext);
130 }
131 
riscv_cpu_has_extension_unlikely(int cpu,const unsigned long ext)132 static __always_inline bool riscv_cpu_has_extension_unlikely(int cpu, const unsigned long ext)
133 {
134 	compiletime_assert(ext < RISCV_ISA_EXT_MAX, "ext must be < RISCV_ISA_EXT_MAX");
135 
136 	if (IS_ENABLED(CONFIG_RISCV_ALTERNATIVE) &&
137 	    __riscv_has_extension_unlikely(STANDARD_EXT, ext))
138 		return true;
139 
140 	return __riscv_isa_extension_available(hart_isa[cpu].isa, ext);
141 }
142 
143 #endif
144