/
miniWeatherML.json
68 lines (68 loc) · 2.56 KB
/
miniWeatherML.json
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
{
"abstract": "miniWeatherML is a machine learning project focused on weather prediction, utilizing advanced numerical methods and high-performance computing resources.",
"description": "miniWeatherML is an open-source project that demonstrates the use of machine learning techniques in weather prediction, leveraging high-performance computing with MPI and CUDA for parallel processing.",
"name": "miniWeatherML",
"homepage": "https://github.com/mrnorman/miniWeatherML",
"references": [],
"requirements": {
"python": ">=3.8",
"pip": [
"hdf5",
"curl",
"zlib",
"netcdf4=1.5.3",
"autoconf",
"automake",
"libtool",
"pkg-config",
"yaml-cpp"
],
"system_modules": [
"gcc/9.5.0",
"cuda/11.8.0",
"cmake/3.26.3",
"openmpi/4.1.2"
]
},
"env_vars": {
"CC": "mpicc",
"FC": "mpif90",
"CXX": "mpic++",
"YAKL_ARCH": "CUDA",
"CMAKE_INCLUDE_PATH": "$CONDA_PREFIX/pnetcdf/include:$CONDA_PREFIX/include",
"CMAKE_LIBRARY_PATH": "$CONDA_PREFIX/pnetcdf/lib:$CONDA_PREFIX/lib",
"YAKL_CUDA_FLAGS": "-DMW_GPU_AWARE_MPI -DHAVE_MPI -O3 -arch sm_70 --use_fast_math -DYAKL_PROFILE -ccbin mpic++",
"YAKL_F90_FLAGS": "-O2 -ffree-line-length-none -DSCREAM_DOUBLE_PRECISION",
"MW_LINK_FLAGS": "-lnetcdf -lpnetcdf"
},
"git": {
"origin": "https://github.com/mrnorman/miniWeatherML.git",
"PnetCDF": "https://github.com/Parallel-NetCDF/PnetCDF.git"
},
"run": {
"community_benchmark": [
"cd {source} && git submodule update --init",
"cd {source}/build && ./cmakescript.sh {source}/experiments/community_benchmark",
"cd {source}/build && make",
"mpirun -np 4 {source}/build/driver {dataset.community_benchmark}"
],
"simple_city": [
"cd {source} && git submodule update --init",
"cd {source}/build && ./cmakescript.sh {source}/experiments/simple_city",
"cd {source}/build && make",
"mpirun -np 4 {source}/build/driver {dataset.simple_city}"
],
"supercell_example": [
"cd {source} && git submodule update --init",
"cd {source}/build && ./cmakescript.sh {source}/experiments/supercell_example",
"cd {source}/build && make",
"mpirun -np 4 {source}/build/driver {dataset.supercell_example}"
],
"supercell_kessler_surrogate": [
"cd {source} && git submodule update --init",
"cd {source}/build && ./cmakescript.sh {source}/experiments/supercell_kessler_surrogate",
"cd {source}/build && make",
"mpirun -np 4 {source}/build/driver {dataset.supercell_kessler_surrogate}"
]
}
}