diff options
| author | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2023-02-02 11:02:55 -0800 |
|---|---|---|
| committer | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2023-02-02 11:02:55 -0800 |
| commit | 493721c40b6fa1e30c24c8fe0782029ab4e53ec6 (patch) | |
| tree | 25366dbb14da709a7d4f181d4cb357a3f4e35a71 /numpy | |
| parent | 9a3568488a19a1a0e4b57fdd17c0e5b3851e46e2 (diff) | |
| download | numpy-493721c40b6fa1e30c24c8fe0782029ab4e53ec6.tar.gz | |
Add cpu test features for avx512fp16
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/core/tests/test_cpu_features.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/numpy/core/tests/test_cpu_features.py b/numpy/core/tests/test_cpu_features.py index 1a76897e2..99f983ce9 100644 --- a/numpy/core/tests/test_cpu_features.py +++ b/numpy/core/tests/test_cpu_features.py @@ -116,7 +116,7 @@ class Test_X86_Features(AbstractTest): "AVX", "F16C", "XOP", "FMA4", "FMA3", "AVX2", "AVX512F", "AVX512CD", "AVX512ER", "AVX512PF", "AVX5124FMAPS", "AVX5124VNNIW", "AVX512VPOPCNTDQ", "AVX512VL", "AVX512BW", "AVX512DQ", "AVX512VNNI", "AVX512IFMA", - "AVX512VBMI", "AVX512VBMI2", "AVX512BITALG", + "AVX512VBMI", "AVX512VBMI2", "AVX512BITALG, AVX512FP16", ] features_groups = dict( AVX512_KNL = ["AVX512F", "AVX512CD", "AVX512ER", "AVX512PF"], @@ -128,6 +128,7 @@ class Test_X86_Features(AbstractTest): "AVX512VBMI"], AVX512_ICL = ["AVX512F", "AVX512CD", "AVX512BW", "AVX512DQ", "AVX512VL", "AVX512IFMA", "AVX512VBMI", "AVX512VNNI", "AVX512VBMI2", "AVX512BITALG", "AVX512VPOPCNTDQ"], + AVX512_SPR = ["AVX512F", "AVX512CD", "AVX512BW", "AVX512DQ", "AVX512VL", "AVX512FP16"], ) features_map = dict( SSE3="PNI", SSE41="SSE4_1", SSE42="SSE4_2", FMA3="FMA", |