Re: [PATCH v3 07/17] riscv: Extend cpufeature.c to detect vendor extensions

On Sat, Apr 20, 2024 at 06:04:39PM -0700, Charlie Jenkins wrote:

Separate vendor extensions out into one struct per vendor instead of adding vendor extensions onto riscv_isa_ext.

Signed-off-by: Charlie Jenkins charlie@rivosinc.com

arch/riscv/Kconfig | 2 + arch/riscv/Kconfig.vendor | 9 +++ arch/riscv/include/asm/cpufeature.h | 18 ++++++ arch/riscv/include/asm/vendor_extensions.h | 26 ++++++++ arch/riscv/include/asm/vendor_extensions/thead.h | 19 ++++++ arch/riscv/kernel/Makefile | 2 + arch/riscv/kernel/cpufeature.c | 75 +++++++++++++++++------- arch/riscv/kernel/vendor_extensions.c | 18 ++++++ arch/riscv/kernel/vendor_extensions/Makefile | 3 + arch/riscv/kernel/vendor_extensions/thead.c | 36 ++++++++++++ 10 files changed, 188 insertions(+), 20 deletions(-)

diff --git a/arch/riscv/Kconfig b/arch/riscv/Kconfig index be09c8836d56..fec86fba3acd 100644 --- a/arch/riscv/Kconfig +++ b/arch/riscv/Kconfig @@ -759,6 +759,8 @@ config RISCV_EFFICIENT_UNALIGNED_ACCESS endchoice +source "arch/riscv/Kconfig.vendor"

endmenu # "Platform type" menu "Kernel features" diff --git a/arch/riscv/Kconfig.vendor b/arch/riscv/Kconfig.vendor new file mode 100644 index 000000000000..d57254f28ea6 --- /dev/null +++ b/arch/riscv/Kconfig.vendor @@ -0,0 +1,9 @@ +config RISCV_ISA_VENDOR_EXT_THEAD

• bool "T-Head vendor extension support"
• default y
• help

•  Say N here if you want to disable all T-Head vendor extension
  

•  support. This will cause any T-Head vendor extensions that are
  

•  requested by hardware probing to be ignored.
  

•  If you don't know what to do here, say Y.
  

diff --git a/arch/riscv/include/asm/cpufeature.h b/arch/riscv/include/asm/cpufeature.h index 809f61ffb667..db6a6d7d6b2e 100644 --- a/arch/riscv/include/asm/cpufeature.h +++ b/arch/riscv/include/asm/cpufeature.h @@ -35,6 +35,24 @@ extern u32 riscv_vlenb_dt[NR_CPUS]; void riscv_user_isa_enable(void); +#define _RISCV_ISA_EXT_DATA(_name, _id, _subset_exts, _subset_exts_size) { \

• .name = #_name, \
• .property = #_name, \
• .id = _id, \
• .subset_ext_ids = _subset_exts, \
• .subset_ext_size = _subset_exts_size \

+}

+#define __RISCV_ISA_EXT_DATA(_name, _id) _RISCV_ISA_EXT_DATA(_name, _id, NULL, 0)

+/* Used to declare pure "lasso" extension (Zk for instance) */ +#define __RISCV_ISA_EXT_BUNDLE(_name, _bundled_exts) \

• _RISCV_ISA_EXT_DATA(_name, RISCV_ISA_EXT_INVALID, _bundled_exts, ARRAY_SIZE(_bundled_exts))

+/* Used to declare extensions that are a superset of other extensions (Zvbb for instance) */ +#define __RISCV_ISA_EXT_SUPERSET(_name, _id, _sub_exts) \

• _RISCV_ISA_EXT_DATA(_name, _id, _sub_exts, ARRAY_SIZE(_sub_exts))

#if defined(CONFIG_RISCV_MISALIGNED) bool check_unaligned_access_emulated_all_cpus(void); void unaligned_emulation_finish(void); diff --git a/arch/riscv/include/asm/vendor_extensions.h b/arch/riscv/include/asm/vendor_extensions.h new file mode 100644 index 000000000000..0af1ddd0af70 --- /dev/null +++ b/arch/riscv/include/asm/vendor_extensions.h @@ -0,0 +1,26 @@ +/* SPDX-License-Identifier: GPL-2.0-only */ +/*

• • Copyright 2024 Rivos, Inc
• */

+#ifndef _ASM_VENDOR_EXTENSIONS_H +#define _ASM_VENDOR_EXTENSIONS_H

+#include <asm/cpufeature.h>

+#include <linux/array_size.h> +#include <linux/types.h>

+struct riscv_isa_vendor_ext_data_list {

• const struct riscv_isa_ext_data *ext_data;
• struct riscv_isainfo *per_hart_vendor_bitmap;
• unsigned long *vendor_bitmap;
• const size_t ext_data_count;
• const size_t bitmap_size;

+};

+extern const struct riscv_isa_vendor_ext_data_list *riscv_isa_vendor_ext_list[];

+extern const size_t riscv_isa_vendor_ext_list_size;

+#endif /* _ASM_VENDOR_EXTENSIONS_H */ diff --git a/arch/riscv/include/asm/vendor_extensions/thead.h b/arch/riscv/include/asm/vendor_extensions/thead.h new file mode 100644 index 000000000000..92eec729888d --- /dev/null +++ b/arch/riscv/include/asm/vendor_extensions/thead.h @@ -0,0 +1,19 @@ +/* SPDX-License-Identifier: GPL-2.0 */ +#ifndef _ASM_RISCV_VENDOR_EXTENSIONS_THEAD_H +#define _ASM_RISCV_VENDOR_EXTENSIONS_THEAD_H

+#include <asm/vendor_extensions.h>

+#include <linux/types.h>

+#define RISCV_ISA_VENDOR_EXT_XTHEADVECTOR 0

+/*

• • Extension keys should be strictly less than max.
• • It is safe to increment this when necessary.
• */

+#define RISCV_ISA_VENDOR_EXT_MAX_THEAD 32

+extern const struct riscv_isa_vendor_ext_data_list riscv_isa_vendor_ext_list_thead;

+#endif diff --git a/arch/riscv/kernel/Makefile b/arch/riscv/kernel/Makefile index 81d94a8ee10f..53361c50fb46 100644 --- a/arch/riscv/kernel/Makefile +++ b/arch/riscv/kernel/Makefile @@ -58,6 +58,8 @@ obj-y += riscv_ksyms.o obj-y += stacktrace.o obj-y += cacheinfo.o obj-y += patch.o +obj-y += vendor_extensions.o +obj-y += vendor_extensions/ obj-y += probes/ obj-y += tests/ obj-$(CONFIG_MMU) += vdso.o vdso/ diff --git a/arch/riscv/kernel/cpufeature.c b/arch/riscv/kernel/cpufeature.c index b537731cadef..c9f36822e337 100644 --- a/arch/riscv/kernel/cpufeature.c +++ b/arch/riscv/kernel/cpufeature.c @@ -24,6 +24,7 @@ #include <asm/processor.h> #include <asm/sbi.h> #include <asm/vector.h> +#include <asm/vendor_extensions.h> #define NUM_ALPHA_EXTS ('z' - 'a' + 1) @@ -102,24 +103,6 @@ static bool riscv_isa_extension_check(int id) return true; } -#define _RISCV_ISA_EXT_DATA(_name, _id, _subset_exts, _subset_exts_size) { \

• .name = #_name, \
• .property = #_name, \
• .id = _id, \
• .subset_ext_ids = _subset_exts, \
• .subset_ext_size = _subset_exts_size \

-}

-#define __RISCV_ISA_EXT_DATA(_name, _id) _RISCV_ISA_EXT_DATA(_name, _id, NULL, 0)

-/* Used to declare pure "lasso" extension (Zk for instance) */ -#define __RISCV_ISA_EXT_BUNDLE(_name, _bundled_exts) \

• _RISCV_ISA_EXT_DATA(_name, RISCV_ISA_EXT_INVALID, _bundled_exts, ARRAY_SIZE(_bundled_exts))

-/* Used to declare extensions that are a superset of other extensions (Zvbb for instance) */ -#define __RISCV_ISA_EXT_SUPERSET(_name, _id, _sub_exts) \

• _RISCV_ISA_EXT_DATA(_name, _id, _sub_exts, ARRAY_SIZE(_sub_exts))

static const unsigned int riscv_zk_bundled_exts[] = { RISCV_ISA_EXT_ZBKB, RISCV_ISA_EXT_ZBKC, @@ -353,6 +336,14 @@ static void __init riscv_parse_isa_string(unsigned long *this_hwcap, struct risc bool ext_long = false, ext_err = false; switch (*ext) {

• case 'x':
  

• case 'X':
  

• 	pr_warn("Vendor extensions are ignored in riscv,isa. Use riscv,isa-extensions instead.");
  

I think I pointed out before that this should only really warn once, not once per extension per cpu. pr_warn_once() please.

• 	/*
  

• 	 * In canonical order, the remaining extensions in the
  

• 	 * isa string will be vendor extensions so exit.
  

• 	 */
  

Yeah, but the binding doesn't require canonical order, so please don't exit & just move along to the next extension.

• 	break;
  

  case 's': /* * Workaround for invalid single-letter 's' & 'u' (QEMU).

@@ -368,8 +359,6 @@ static void __init riscv_parse_isa_string(unsigned long *this_hwcap, struct risc } fallthrough; case 'S':

• case 'x':
  

• case 'X':
  

  case 'z': case 'Z': /*

@@ -580,6 +569,48 @@ static void __init riscv_fill_hwcap_from_isa_string(unsigned long *isa2hwcap) acpi_put_table((struct acpi_table_header *)rhct); } +static void __init riscv_add_cpu_vendor_ext(struct device_node *cpu_node, int cpu)

Can we have a consistent naming here? The other functions are using "fill".

+{

• for (int i = 0; i < riscv_isa_vendor_ext_list_size; i++) {

• const struct riscv_isa_vendor_ext_data_list *ext_list = riscv_isa_vendor_ext_list[i];
  

• for (int j = 0; j < ext_list->ext_data_count; j++) {
  

• 	const struct riscv_isa_ext_data ext = ext_list->ext_data[j];
  

• 	struct riscv_isainfo *isavendorinfo = &ext_list->per_hart_vendor_bitmap[cpu];
  

• 	if (of_property_match_string(cpu_node, "riscv,isa-extensions",
  

• 				     ext.property) < 0)
  

• 		continue;
  

• 	/*
  

• 	 * Assume that subset extensions are all members of the
  

• 	 * same vendor.
  

• 	 */
  

• 	if (ext.subset_ext_size)
  

• 		for (int k = 0; k < ext.subset_ext_size; k++)
  

• 			set_bit(ext.subset_ext_ids[k], isavendorinfo->isa);
  

• 	set_bit(ext.id, isavendorinfo->isa);
  

• }
  

• }

+}

+static void __init set_riscv_isa_vendor(int cpu)

Again, consistent naming here would have riscv_isa_set... And it's not setting a vendor, can you pick a name that matches the action that the function is actually taking please? riscv_isa_set_vendor_bitmask() or w/e.

Also, you've added Kconfig options for the different vendors but this code is unconditionally compiled. Do we need to have a hidden Kconfig option that all of the vendor options select that gets all of this new code culled when none of them are enabled? Could just be as simple as if (!IS_ENABLED(CONFIG_FOO)) return; in these functions in cpufeature.c?

+{

• for (int i = 0; i < riscv_isa_vendor_ext_list_size; i++) {

• const struct riscv_isa_vendor_ext_data_list *ext_list = riscv_isa_vendor_ext_list[i];
  

• if (bitmap_empty(ext_list->vendor_bitmap, ext_list->bitmap_size))
  

• 	bitmap_copy(ext_list->vendor_bitmap,
  

• 		    ext_list->per_hart_vendor_bitmap[cpu].isa,
  

• 		    ext_list->bitmap_size);
  

• else
  

• 	bitmap_and(ext_list->vendor_bitmap, ext_list->vendor_bitmap,
  

• 		   ext_list->per_hart_vendor_bitmap[cpu].isa,
  

• 		   ext_list->bitmap_size);
  

• }

+}

static int __init riscv_fill_hwcap_from_ext_list(unsigned long *isa2hwcap) { unsigned int cpu; @@ -623,6 +654,8 @@ static int __init riscv_fill_hwcap_from_ext_list(unsigned long *isa2hwcap) } }

• riscv_add_cpu_vendor_ext(cpu_node, cpu);
  

• of_node_put(cpu_node);

/* @@ -638,6 +671,8 @@ static int __init riscv_fill_hwcap_from_ext_list(unsigned long *isa2hwcap) bitmap_copy(riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX); else bitmap_and(riscv_isa, riscv_isa, isainfo->isa, RISCV_ISA_EXT_MAX);

• set_riscv_isa_vendor(cpu);
  

  }

if (bitmap_empty(riscv_isa, RISCV_ISA_EXT_MAX)) diff --git a/arch/riscv/kernel/vendor_extensions.c b/arch/riscv/kernel/vendor_extensions.c new file mode 100644 index 000000000000..f76cb3013c2d --- /dev/null +++ b/arch/riscv/kernel/vendor_extensions.c @@ -0,0 +1,18 @@ +// SPDX-License-Identifier: GPL-2.0-only +/*

• • Copyright 2024 Rivos, Inc
• */

+#include <asm/vendor_extensions.h> +#include <asm/vendor_extensions/thead.h>

+#include <linux/array_size.h> +#include <linux/types.h>

+const struct riscv_isa_vendor_ext_data_list *riscv_isa_vendor_ext_list[] = { +#ifdef CONFIG_RISCV_ISA_VENDOR_EXT_THEAD

• &riscv_isa_vendor_ext_list_thead,

+#endif +};

+const size_t riscv_isa_vendor_ext_list_size = ARRAY_SIZE(riscv_isa_vendor_ext_list); diff --git a/arch/riscv/kernel/vendor_extensions/Makefile b/arch/riscv/kernel/vendor_extensions/Makefile new file mode 100644 index 000000000000..3383066baaab --- /dev/null +++ b/arch/riscv/kernel/vendor_extensions/Makefile @@ -0,0 +1,3 @@ +# SPDX-License-Identifier: GPL-2.0-only

+obj-$(CONFIG_RISCV_ISA_VENDOR_EXT_THEAD) += thead.o diff --git a/arch/riscv/kernel/vendor_extensions/thead.c b/arch/riscv/kernel/vendor_extensions/thead.c new file mode 100644 index 000000000000..a0a47414ed22 --- /dev/null +++ b/arch/riscv/kernel/vendor_extensions/thead.c @@ -0,0 +1,36 @@ +// SPDX-License-Identifier: GPL-2.0-only

+#include <asm/cpufeature.h> +#include <asm/vendor_extensions.h> +#include <asm/vendor_extensions/thead.h>

+#include <linux/array_size.h> +#include <linux/types.h>

+/* All T-Head vendor extensions supported in Linux */ +const struct riscv_isa_ext_data riscv_isa_vendor_ext_thead[] = {

• _RISCV_ISA_EXT_DATA(xtheadvector, RISCV_ISA_VENDOR_EXT_XTHEADVECTOR, NULL, 0),

+};

+/*

• • The first member of this struct must be a bitmap named isa so it can be
• • compatible with riscv_isainfo even though the sizes of the bitmaps may be
• • different.
• */

+struct riscv_isavendorinfo_thead {

• DECLARE_BITMAP(isa, RISCV_ISA_VENDOR_EXT_MAX_THEAD);

+};

+/* Hart specific T-Head vendor extension support */ +static struct riscv_isavendorinfo_thead hart_vendorinfo_thead[NR_CPUS];

+/* Set of T-Head vendor extensions supported on all harts */ +DECLARE_BITMAP(vendorinfo_thead, RISCV_ISA_VENDOR_EXT_MAX_THEAD);

+const struct riscv_isa_vendor_ext_data_list riscv_isa_vendor_ext_list_thead = {

• .ext_data = riscv_isa_vendor_ext_thead,
• .per_hart_vendor_bitmap = (struct riscv_isainfo *)hart_vendorinfo_thead,
• .vendor_bitmap = vendorinfo_thead,
• .ext_data_count = ARRAY_SIZE(riscv_isa_vendor_ext_thead),
• .bitmap_size = RISCV_ISA_VENDOR_EXT_MAX_THEAD

+};

-- 2.44.0