Skip to content

Commit 9ad0d80

Browse files
authored
fix: icon uenv for a100 (#75)
* fix icon uenv for a100 building with c2sm-spack * add cuda_arch to packages.yaml with post-install script
1 parent 883ab9d commit 9ad0d80

File tree

2 files changed

+15
-3
lines changed

2 files changed

+15
-3
lines changed

Diff for: recipes/wcp/icon/v1/a100/environments.yaml

+4-3
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,11 @@ nvidia:
1414
- [email protected]%nvhpc +tools +fortran +aec +openmp jp2k=jasper
1515
- cmake%gcc
1616
17-
- hdf5%gcc
17+
- openblas%gcc
1818
- hwloc%gcc
19-
- netcdf-c%gcc
20-
- netcdf-cxx4%gcc
19+
- hdf5+szip+fortran%nvhpc
20+
- netcdf-c%nvhpc
21+
- netcdf-cxx4%nvhpc
2122
- netcdf-fortran%nvhpc
2223
- numactl%gcc
2324

Diff for: recipes/wcp/icon/v1/a100/post-install

+11
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
#!/usr/bin/python3
2+
3+
import yaml
4+
5+
with open("{{ env.mount }}/config/packages.yaml", mode='r') as file:
6+
packages = yaml.safe_load(file)
7+
8+
packages['packages']['all'] = {'variants': 'cuda_arch=80'}
9+
10+
with open("{{ env.mount }}/config/packages.yaml", mode='w') as file:
11+
yaml.dump(packages, file)

0 commit comments

Comments
 (0)