Skip to content

C3SE numpy updates #5

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

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

C3SE numpy updates #5

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
170fa6b
numpy_check: remove usage of find_modules for numpy check
yqshao Mar 11, 2025
ec577fe
numpy_check: check in some figures for cascadelake and icelake
yqshao Mar 11, 2025
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
45 changes: 30 additions & 15 deletions checks/apps/python/numpy_check.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,10 @@
from hpctestlib.python.numpy.numpy_ops import numpy_ops_check
from reframe.utility import find_modules

class snic_numpy_test(numpy_ops_check):
@rfm.simple_test
class numpy_check(numpy_ops_check):
valid_prog_environs = ['builtin']
valid_systems = ['alvis', 'kebnekaise']
num_tasks_per_node = 1
use_multithreading = False
all_ref = {
Expand Down Expand Up @@ -41,6 +43,20 @@ class snic_numpy_test(numpy_ops_check):
'eigendec': (4.14, None, 0.05, 's'),
'inv': (0.16, None, 0.05, 's'),
},
'cascadelake@16c': {
'dot': (0.16, None, 0.05, 's'),
'svd': (0.64, None, 0.05, 's'),
'cholesky': (0.13, None, 0.05, 's'),
'eigendec': (4.96, None, 0.05, 's'),
'inv': (0.21, None, 0.05, 's'),
},
'cascadelake@32c': {
'dot': (0.16, None, 0.05, 's'),
'svd': (0.64, None, 0.05, 's'),
'cholesky': (0.13, None, 0.05, 's'),
'eigendec': (4.96, None, 0.05, 's'),
'inv': (0.21, None, 0.05, 's'),
},
'skylake@14c': {
'dot': (0.3, None, 0.05, 's'),
'svd': (0.35, None, 0.05, 's'),
Expand All @@ -55,15 +71,24 @@ class snic_numpy_test(numpy_ops_check):
'eigendec': (4.14, None, 0.05, 's'),
'inv': (0.16, None, 0.05, 's'),
},
'icelake@64c': {
'dot': (0.19, None, 0.05, 's'),
'svd': (1.29, None, 0.05, 's'),
'cholesky': (0.46, None, 0.05, 's'),
'eigendec': (9.56, None, 0.05, 's'),
'inv': (0.36, None, 0.05, 's'),
},
}
tags = {'production'}
maintainers = ['RS', 'TR', 'AS']

@run_after('init')
def process_module_info(self):
s, e, m = self.module_info
self.valid_prog_environs = [e]
self.modules = [m]
def set_modules(self):
module = {
'kebnekaise': [],
'alvis': ['SciPy-bundle/2024.05-gfbf-2024a'],
}
self.modules = module.get(self.current_system.name, [])

@run_after('setup')
def set_num_cpus_per_task(self):
Expand All @@ -80,13 +105,3 @@ def set_perf_ref(self):
self.reference = {
pname: self.all_ref[f'{arch}@{num_cores}c']
}

@rfm.simple_test
class c3se_numpy_test(snic_numpy_test):
module_info = parameter(find_modules('SciPy-bundle', { r'.*': 'builtin' }))
valid_systems = ['alvis']

@rfm.simple_test
class hpc2n_numpy_test(snic_numpy_test):
module_info = parameter(find_modules('SciPy-bundle'))
valid_systems = ['kebnekaise']