ramble/examples/tutorial_11_exec_order_config.yaml at develop · GoogleCloudPlatform/ramble · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# Copyright 2022-2026 The Ramble Authors

# Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
# https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
# <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
# option. This file may not be copied, modified, or distributed
# except according to those terms.

ramble:
  variants:
    package_manager: spack
  env_vars:
    set:
      OMP_NUM_THREADS: '{n_threads}'
  variables:
    processes_per_node: 16
    n_ranks: '{processes_per_node}*{n_nodes}'
    batch_submit: '{execute_experiment}'
    mpi_command: mpirun -n {n_ranks}
  applications:
    wrf@4.2:
      workloads:
        CONUS_12km:
          experiments:
            scaling_{n_nodes}:
              internals:
                custom_executables:
                  start_time:
                    template:
                    - date +%s
                    redirect: '{experiment_run_dir}/start_time'
                    use_mpi: false
                  end_time:
                    template:
                    - date +%s
                    redirect: '{experiment_run_dir}/end_time'
                    use_mpi: false
                executables:
                - builtin::env_vars
                - package_manager_builtin::spack::spack_source
                - package_manager_builtin::spack::spack_activate
                - stage-files
                - stage-namelist
                - cleanup
                - define_nproc_y
                - define_nproc_x
                - fix_12km
                - start_time
                - execute
                - end_time
                - copy-logs
                - post-exec-clean
              variables:
                n_nodes: [1, 2]
  software:
    packages:
      gcc14:
        pkg_spec: gcc@14.2.0
      intel-mpi:
        pkg_spec: intel-oneapi-mpi@2021.17.2
        compiler: gcc14
      wrfv4-{application::wrf::version}:
        pkg_spec: wrf@{application::wrf::version} build_type=dm+sm compile_type=em_real
          nesting=basic ~pnetcdf
        compiler: gcc14
    environments:
      wrf@4.2:
        packages:
        - intel-mpi
        - wrfv4-{application::wrf::version}