你已经派生过 gprMax
镜像自地址
https://gitee.com/sunhf/gprMax.git
已同步 2025-08-07 23:14:03 +08:00
48 行
1.4 KiB
Python
48 行
1.4 KiB
Python
import reframe as rfm
|
|
from reframe.core.builtins import parameter, run_after
|
|
|
|
from reframe_tests.tests.base_tests import GprMaxRegressionTest
|
|
|
|
"""ReFrame tests for performance benchmarking
|
|
|
|
Usage:
|
|
cd gprMax/reframe_tests
|
|
reframe -C configuraiton/{CONFIG_FILE} -c reframe_benchmarks.py -c base_tests.py -r
|
|
"""
|
|
|
|
|
|
@rfm.simple_test
|
|
class SingleNodeBenchmark(GprMaxRegressionTest):
|
|
tags = {"benchmark", "single node", "openmp"}
|
|
|
|
omp_threads = parameter([1, 2, 4, 8, 16, 32, 64, 128])
|
|
# domain = parameter([0.1, 0.15, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8])
|
|
cpu_freq = parameter([2000000, 2250000])
|
|
time_limit = "8h"
|
|
sourcesdir = "src"
|
|
model = parameter(
|
|
[
|
|
"benchmark_model_10",
|
|
"benchmark_model_15",
|
|
"benchmark_model_20",
|
|
"benchmark_model_30",
|
|
"benchmark_model_40",
|
|
"benchmark_model_50",
|
|
"benchmark_model_60",
|
|
"benchmark_model_70",
|
|
"benchmark_model_80",
|
|
]
|
|
)
|
|
|
|
@run_after("init")
|
|
def setup_env_vars(self):
|
|
self.num_cpus_per_task = self.omp_threads
|
|
self.env_vars["SLURM_CPU_FREQ_REQ"] = self.cpu_freq
|
|
super().setup_env_vars()
|
|
|
|
@run_after("init")
|
|
def set_model_file(self):
|
|
input_file = f"benchmark_model_{self.domain}.in"
|
|
self.executable_opts = [input_file]
|
|
self.keep_files = [input_file]
|