<sys/platform/x86.h>: Add AVX-VNNI-INT8 support

Add AVX-VNNI-INT8 support to <sys/platform/x86.h>.
Reviewed-by: Noah Goldstein <goldstein.w.n@gmail.com>
This commit is contained in:
H.J. Lu 2023-04-05 09:21:41 -07:00
parent b39741b45f
commit 4c120c88a6
5 changed files with 20 additions and 0 deletions

View File

@ -225,6 +225,9 @@ Leaf (EAX = 23H).
@item
@code{AVX_VNNI} -- The AVX-VNNI instruction extensions.
@item
@code{AVX_VNNI_INT8} -- The AVX-VNNI-INT8 instruction extensions.
@item
@code{AVX512_4FMAPS} -- The AVX512_4FMAPS instruction extensions.

View File

@ -304,6 +304,12 @@ enum
x86_cpu_LAM = x86_cpu_index_7_ecx_1_eax + 26,
x86_cpu_MSRLIST = x86_cpu_index_7_ecx_1_eax + 27,
x86_cpu_index_7_ecx_1_edx
= (CPUID_INDEX_7_ECX_1 * 8 * 4 * sizeof (unsigned int)
+ cpuid_register_index_edx * 8 * sizeof (unsigned int)),
x86_cpu_AVX_VNNI_INT8 = x86_cpu_index_7_ecx_1_edx + 4,
x86_cpu_index_19_ebx
= (CPUID_INDEX_19 * 8 * 4 * sizeof (unsigned int)
+ cpuid_register_index_ebx * 8 * sizeof (unsigned int)),

View File

@ -143,6 +143,8 @@ update_active (struct cpu_features *cpu_features)
CPU_FEATURE_SET_ACTIVE (cpu_features, AVX_IFMA);
/* Determine if AVX-VNNI is usable. */
CPU_FEATURE_SET_ACTIVE (cpu_features, AVX_VNNI);
/* Determine if AVX-VNNI-INT8 is usable. */
CPU_FEATURE_SET_ACTIVE (cpu_features, AVX_VNNI_INT8);
/* Determine if FMA is usable. */
CPU_FEATURE_SET_ACTIVE (cpu_features, FMA);
/* Determine if VAES is usable. */

View File

@ -314,6 +314,9 @@ enum
#define bit_cpu_AVX_IFMA (1u << 23)
#define bit_cpu_LAM (1u << 26)
/* EDX. */
#define bit_cpu_AVX_VNNI_INT8 (1u << 4)
/* CPUID_INDEX_19. */
/* EBX. */
@ -552,6 +555,7 @@ enum
#define index_cpu_HRESET CPUID_INDEX_7_ECX_1
#define index_cpu_AVX_IFMA CPUID_INDEX_7_ECX_1
#define index_cpu_LAM CPUID_INDEX_7_ECX_1
#define index_cpu_AVX_VNNI_INT8 CPUID_INDEX_7_ECX_1
/* CPUID_INDEX_19. */
@ -792,6 +796,9 @@ enum
#define reg_AVX_IFMA eax
#define reg_LAM eax
/* EDX. */
#define reg_AVX_VNNI_INT8 edx
/* CPUID_INDEX_19. */
/* EBX. */

View File

@ -215,6 +215,7 @@ do_test (void)
CHECK_CPU_FEATURE_PRESENT (AVX_IFMA);
CHECK_CPU_FEATURE_PRESENT (LAM);
CHECK_CPU_FEATURE_PRESENT (MSRLIST);
CHECK_CPU_FEATURE_PRESENT (AVX_VNNI_INT8);
CHECK_CPU_FEATURE_PRESENT (AESKLE);
CHECK_CPU_FEATURE_PRESENT (WIDE_KL);
CHECK_CPU_FEATURE_PRESENT (PTWRITE);
@ -382,6 +383,7 @@ do_test (void)
CHECK_CPU_FEATURE_ACTIVE (FSRCS);
CHECK_CPU_FEATURE_ACTIVE (AMX_FP16);
CHECK_CPU_FEATURE_ACTIVE (AVX_IFMA);
CHECK_CPU_FEATURE_ACTIVE (AVX_VNNI_INT8);
CHECK_CPU_FEATURE_ACTIVE (AESKLE);
CHECK_CPU_FEATURE_ACTIVE (WIDE_KL);
CHECK_CPU_FEATURE_ACTIVE (PTWRITE);