From d88104fd0e2f85ac4eb667fcd7161aaa1d41e11a Mon Sep 17 00:00:00 2001 From: Rafael Cardoso Fernandes Sousa Date: Fri, 7 Jan 2022 10:48:36 -0600 Subject: ENH: Add CPU feature detection for VSX4 (Power10) --- numpy/distutils/ccompiler_opt.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'numpy/distutils/ccompiler_opt.py') 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 -- cgit v1.2.1 From 78c6713b4ee06bec1fa579f94b466cbe5fe4e8f3 Mon Sep 17 00:00:00 2001 From: Rafael Cardoso Fernandes Sousa Date: Fri, 14 Jan 2022 06:05:34 -0600 Subject: STY: Fix linter issue --- numpy/distutils/ccompiler_opt.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'numpy/distutils/ccompiler_opt.py') diff --git a/numpy/distutils/ccompiler_opt.py b/numpy/distutils/ccompiler_opt.py index 794d88433..854584998 100644 --- a/numpy/distutils/ccompiler_opt.py +++ b/numpy/distutils/ccompiler_opt.py @@ -295,7 +295,8 @@ class _Config: ## 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), + 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"), -- cgit v1.2.1