mirror/glibc
2
0
Fork 0
mirror of git://sourceware.org/git/glibc.git synced 2025-03-13 13:37:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

<sys/platform/x86.h>: Add AMX-COMPLEX support

Browse Source 
Add AMX-COMPLEX 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:43 -07:00
parent 609b7b2d3c
commit b05521c916
5 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
manual/platform.texi
View File

@ -197,6 +197,9 @@ The supported processor features are:
@item
@code{AMX_BF16} -- Tile computational operations on bfloat16 numbers.
@item
@code{AMX_COMPLEX} -- Tile computational operations on complex FP16 numbers.
@item
@code{AMX_INT8} -- Tile computational operations on 8-bit numbers.

1
sysdeps/x86/bits/platform/x86.h
View File

@ -310,6 +310,7 @@ enum
x86_cpu_AVX_VNNI_INT8 = x86_cpu_index_7_ecx_1_edx + 4,
x86_cpu_AVX_NE_CONVERT = x86_cpu_index_7_ecx_1_edx + 5,
x86_cpu_AMX_COMPLEX = x86_cpu_index_7_ecx_1_edx + 8,
x86_cpu_index_19_ebx
= (CPUID_INDEX_19 * 8 * 4 * sizeof (unsigned int)

2
sysdeps/x86/cpu-features.c
View File

@ -221,6 +221,8 @@ update_active (struct cpu_features *cpu_features)
CPU_FEATURE_SET_ACTIVE (cpu_features, AMX_INT8);
/* Determine if AMX_FP16 is usable. */
CPU_FEATURE_SET_ACTIVE (cpu_features, AMX_FP16);
/* Determine if AMX_COMPLEX is usable. */
CPU_FEATURE_SET_ACTIVE (cpu_features, AMX_COMPLEX);
}
/* These features are usable only when OSXSAVE is enabled. */

3
sysdeps/x86/include/cpu-features.h
View File

@ -317,6 +317,7 @@ enum
/* EDX. */
#define bit_cpu_AVX_VNNI_INT8 (1u << 4)
#define bit_cpu_AVX_NE_CONVERT (1u << 5)
#define bit_cpu_AMX_COMPLEX (1u << 8)
/* CPUID_INDEX_19. */
@ -558,6 +559,7 @@ enum
#define index_cpu_LAM CPUID_INDEX_7_ECX_1
#define index_cpu_AVX_VNNI_INT8 CPUID_INDEX_7_ECX_1
#define index_cpu_AVX_NE_CONVERT CPUID_INDEX_7_ECX_1
#define index_cpu_AMX_COMPLEX CPUID_INDEX_7_ECX_1
/* CPUID_INDEX_19. */
@ -801,6 +803,7 @@ enum
/* EDX. */
#define reg_AVX_VNNI_INT8 edx
#define reg_AVX_NE_CONVERT edx
#define reg_AMX_COMPLEX edx
/* CPUID_INDEX_19. */

2
sysdeps/x86/tst-get-cpu-features.c
View File

@ -217,6 +217,7 @@ do_test (void)
CHECK_CPU_FEATURE_PRESENT (MSRLIST);
CHECK_CPU_FEATURE_PRESENT (AVX_VNNI_INT8);
CHECK_CPU_FEATURE_PRESENT (AVX_NE_CONVERT);
CHECK_CPU_FEATURE_PRESENT (AMX_COMPLEX);
CHECK_CPU_FEATURE_PRESENT (AESKLE);
CHECK_CPU_FEATURE_PRESENT (WIDE_KL);
CHECK_CPU_FEATURE_PRESENT (PTWRITE);
@ -386,6 +387,7 @@ do_test (void)
CHECK_CPU_FEATURE_ACTIVE (AVX_IFMA);
CHECK_CPU_FEATURE_ACTIVE (AVX_VNNI_INT8);
CHECK_CPU_FEATURE_ACTIVE (AVX_NE_CONVERT);
CHECK_CPU_FEATURE_ACTIVE (AMX_COMPLEX);
CHECK_CPU_FEATURE_ACTIVE (AESKLE);
CHECK_CPU_FEATURE_ACTIVE (WIDE_KL);
CHECK_CPU_FEATURE_ACTIVE (PTWRITE);