diff options
| author | Matti Picus <matti.picus@gmail.com> | 2022-01-18 19:11:41 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-18 19:11:41 +0200 |
| commit | f9c4596e54a3810c5a398438f914091777a1b73b (patch) | |
| tree | 268380edd49632f4f341dcf146816b030b44b94b /numpy/distutils/ccompiler_opt.py | |
| parent | 72396eace8d20f48716900886ab2531b249f42ae (diff) | |
| parent | 3784cac6f79b1a2f5828957173cac81bbc483ba3 (diff) | |
| download | numpy-f9c4596e54a3810c5a398438f914091777a1b73b.tar.gz | |
Merge pull request #20821 from rafaelcfsousa/p10_enablement
ENH: Add CPU feature detection for POWER10 (VSX4)
Diffstat (limited to 'numpy/distutils/ccompiler_opt.py')
| -rw-r--r-- | numpy/distutils/ccompiler_opt.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/numpy/distutils/ccompiler_opt.py b/numpy/distutils/ccompiler_opt.py index f1d024b94..854584998 100644 --- a/numpy/distutils/ccompiler_opt.py +++ b/numpy/distutils/ccompiler_opt.py @@ -294,6 +294,9 @@ 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", implies_detect=False, + extra_checks="VSX4_MMA"), # IBM/Z ## VX(z13) support VX = dict(interest=1, headers="vecintrin.h"), @@ -471,12 +474,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 |