1
1
2
- from machines_specs import assert_machine_supported , \
3
- get_mach_cxx_compiler , get_mach_c_compiler , get_mach_f90_compiler , \
4
- get_mach_batch_command , get_mach_env_setup_command , \
5
- get_mach_baseline_root_dir , is_cuda_machine , \
6
- get_mach_compilation_resources , get_mach_testing_resources
2
+ from machines_specs import assert_machine_supported , get_machine , get_mach_env_setup_command
7
3
from utils import expect
8
4
9
5
CHOICES = (
@@ -24,23 +20,24 @@ def query_scream(machine, param):
24
20
assert_machine_supported (machine )
25
21
expect (param in CHOICES , f"Unknown param { param } " )
26
22
23
+ mach = get_machine (machine )
27
24
if param == "cxx_compiler" :
28
- return get_mach_cxx_compiler ( machine )
25
+ return mach . cxx_compiler
29
26
elif param == "c_compiler" :
30
- return get_mach_c_compiler ( machine )
27
+ return mach . c_compiler
31
28
elif param == "f90_compiler" :
32
- return get_mach_f90_compiler ( machine )
29
+ return mach . ftn_compiler
33
30
elif param == "batch" :
34
- return get_mach_batch_command ( machine )
31
+ return mach . batch
35
32
elif param == "env" :
36
33
return get_mach_env_setup_command (machine )
37
34
elif param == "baseline_root" :
38
- return get_mach_baseline_root_dir ( machine )
35
+ return mach . baselines_dir
39
36
elif param == "cuda" :
40
- return str (is_cuda_machine ( machine ) )
37
+ return str (mach . gpu_arch == "cuda" )
41
38
elif param == "comp_j" :
42
- return get_mach_compilation_resources ()
39
+ return num_bld_res
43
40
elif param == "test_j" :
44
- return get_mach_testing_resources ( machine )
41
+ return gnum_run_res
45
42
else :
46
43
expect (False , f"Unhandled param { param } " )
0 commit comments