Skip to content

Commit

Permalink
deploy QE on todi (#114)
Browse files Browse the repository at this point in the history
  • Loading branch information
simonpintarelli authored Jul 2, 2024
1 parent 1261913 commit 0a4362b
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 9 deletions.
1 change: 1 addition & 0 deletions config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,7 @@ uenvs:
gh200: v7.3.1/gh200
deploy:
santis: [gh200]
todi: [gh200]
sirius:
"v7.5":
recipes:
Expand Down
15 changes: 6 additions & 9 deletions recipes/quantumespresso/v7.3.1/gh200/environments.yaml
Original file line number Diff line number Diff line change
@@ -1,23 +1,20 @@
nvhpc-env:
compiler:
- toolchain: llvm
spec: nvhpc
- toolchain: gcc
spec: gcc@12
- toolchain: llvm
spec: nvhpc
mpi:
spec: [email protected]
spec: [email protected]%nvhpc
gpu: cuda
unify: true
specs:
- [email protected] %nvhpc +libxc +cuda +nvtx +scalapack
- cmake%gcc
- fftw%gcc
- netlib-scalapack%gcc
- [email protected] %nvhpc +libxc +cuda +nvtx +scalapack +openmp
- netlib-scalapack
- fftw%nvhpc
- nvpl-blas threads=openmp
- nvpl-lapack threads=openmp
- libxc%nvhpc~cuda
- libxml2%gcc
- patchelf%gcc
packages:
- git
variants:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -296,6 +296,11 @@ class QuantumEspresso(CMakePackage, Package):

conflicts("@6.5:", when="+environ", msg="6.4.x is the latest QE series supported by Environ")

# 7.3 - a compile-time problem fixed in 7.3.1
patch_url = "https://gitlab.com/QEF/q-e/-/commit/b98ff7539e5731728d2d49ac01021a57f2594027.diff"
patch_checksum = "04c125d249d1f076abe04bc4de39bd3b44a41a78d6233b638a17bd96f91443d5"
patch(patch_url, sha256=patch_checksum, when="@=7.3+elpa build_system=cmake")

# QE 7.1 fix post-processing install part 1/2
# see: https://gitlab.com/QEF/q-e/-/merge_requests/2005
patch_url = "https://gitlab.com/QEF/q-e/-/commit/4ca3afd4c6f27afcf3f42415a85a353a7be1bd37.diff"
Expand Down Expand Up @@ -467,6 +472,14 @@ def cmake_args(self):
if spec.satisfies("@:7.1"):
cmake_args.append(self.define("BLA_VENDOR", "All"))

if "^nvpl-blas" in spec and "^nvpl-lapack" in spec:
cmake_args.append(self.define("BLAS_LIBRARIES", spec["blas"].libs.joined(";")))
cmake_args.append(self.define("LAPACK_LIBRARIES", spec["lapack"].libs.joined(";")))
# Up to [email protected] set BLA_VENDOR to All to force detection of vanilla scalapack
if spec.satisfies("@:7.1"):
cmake_args.append(self.define("BLA_VENDOR", "All"))


if plugins:
cmake_args.append(self.define("QE_ENABLE_PLUGINS", plugins))
return cmake_args
Expand Down

0 comments on commit 0a4362b

Please sign in to comment.