Skip to content

TST: fix POWER VSX feature mapping #30801

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
charris merged 6 commits into from
Mar 26, 2026
+33 −1
Merged

TST: fix POWER VSX feature mapping #30801

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
0576d7c
fix POWER VSX test feature mapping
abslock128w Feb 8, 2026
4f671d8
treat VSX as baseline in CPU feature tests
abslock128w Feb 8, 2026
e9376e8
fix POWER VSX feature detection using platform inference
abslock128w Feb 8, 2026
330acf0
Black space formatting
abslock128w Feb 9, 2026
2d95860
removed the one trailing space i fogot about+_+
abslock128w Feb 11, 2026
0c4124b
Update test_cpu_features.py
abslock128w Feb 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix POWER VSX feature detection using platform inference 
Update features map and load_flags method for POWER architecture detection.
  • Loading branch information
@abslock128w
abslock128w authored Feb 8, 2026
commit e9376e8c8adabd36fdea057c22e2256197ffa6ee
35 changes: 33 additions & 2 deletions numpy/_core/tests/test_cpu_features.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -387,11 +387,42 @@ def load_flags(self):
@pytest.mark.skipif(not is_linux or not is_power, reason="Only for Linux and Power")
class Test_POWER_Features(AbstractTest):
features = ["VSX", "VSX2", "VSX3", "VSX4"]
features_map = { "VSX": "ARCH_2_06", "VSX2": "ARCH_2_07", "VSX3": "ARCH_3_00", "VSX4": "ARCH_3_1"}
features_map = {
"VSX": "ARCH_2_06",
"VSX2": "ARCH_2_07",
"VSX3": "ARCH_3_00",
"VSX4": "ARCH_3_1B"
}

def load_flags(self):
self.load_flags_auxv()
self.features_flags.add("ARCH_2_06")
platform = self._get_platform()

if platform:
power_match = re.search(r'power(\d+)', platform, re.IGNORECASE)
if power_match:
power_gen = int(power_match.group(1))
if power_gen >= 7:
self.features_flags.add("ARCH_2_06")
if power_gen >= 8:
self.features_flags.add("ARCH_2_07")
if power_gen >= 9:
self.features_flags.add("ARCH_3_00")
if power_gen >= 10:
self.features_flags.add("ARCH_3_1B")

def _get_platform(self):
"""Get the AT_PLATFORM value from AUXV"""
try:
auxv = subprocess.check_output(['/bin/true'], env={"LD_SHOW_AUXV": "1"})
for line in auxv.split(b'\n'):
if line.startswith(b'AT_PLATFORM'):
parts = line.split(b':', 1)
if len(parts) == 2:
return parts[1].strip().decode().lower()
except Exception:
pass
return None


is_zarch = re.match(r"^(s390x)", machine, re.IGNORECASE)
Expand Down
Loading