diff options
| author | Rafael Cardoso Fernandes Sousa <rafaelcfsousa@ibm.com> | 2022-01-07 10:48:36 -0600 |
|---|---|---|
| committer | Rafael Cardoso Fernandes Sousa <rafaelcfsousa@ibm.com> | 2022-01-13 14:47:51 -0600 |
| commit | d88104fd0e2f85ac4eb667fcd7161aaa1d41e11a (patch) | |
| tree | dfcfeafda07a79d06d75bbaa104ebb7e2d51ced0 /numpy/distutils/ccompiler_opt.py | |
| parent | 7191d9a4773d77205349ac151f84b72c0ffcf848 (diff) | |
| download | numpy-d88104fd0e2f85ac4eb667fcd7161aaa1d41e11a.tar.gz | |
ENH: Add CPU feature detection for VSX4 (Power10)
Diffstat (limited to 'numpy/distutils/ccompiler_opt.py')
| -rw-r--r-- | numpy/distutils/ccompiler_opt.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/distutils/ccompiler_opt.py b/numpy/distutils/ccompiler_opt.py index f1d024b94..794d88433 100644 --- a/numpy/distutils/ccompiler_opt.py +++ b/numpy/distutils/ccompiler_opt.py @@ -294,6 +294,8 @@ class _Config: VSX2 = dict(interest=2, implies="VSX", implies_detect=False), ## Power9/ISA 3.00 VSX3 = dict(interest=3, implies="VSX2", implies_detect=False), + ## Power10/ISA 3.1 + VSX4 = dict(interest=4, implies="VSX3", extra_checks="VSX4_MMA", implies_detect=False), # IBM/Z ## VX(z13) support VX = dict(interest=1, headers="vecintrin.h"), @@ -471,12 +473,16 @@ class _Config: ), VSX3 = dict( flags="-mcpu=power9 -mtune=power9", implies_detect=False + ), + VSX4 = dict( + flags="-mcpu=power10 -mtune=power10", implies_detect=False ) ) if self.cc_is_clang: partial["VSX"]["flags"] = "-maltivec -mvsx" partial["VSX2"]["flags"] = "-mpower8-vector" partial["VSX3"]["flags"] = "-mpower9-vector" + partial["VSX4"]["flags"] = "-mpower10-vector" return partial |