diff options
| author | Anton Kozhevnikov <toxa81@gmail.com> | 2023-08-31 16:16:31 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-31 16:16:31 +0200 |
| commit | 679d41ea669c1a435c65447dee2b764623d6c318 (patch) | |
| tree | 435691320844ca5493f672776021f9fead92fb58 | |
| parent | 86216cc36ed8ea2771a71f1ea01c18a10f368afa (diff) | |
| download | spack-679d41ea669c1a435c65447dee2b764623d6c318.tar.gz spack-679d41ea669c1a435c65447dee2b764623d6c318.tar.bz2 spack-679d41ea669c1a435c65447dee2b764623d6c318.tar.xz spack-679d41ea669c1a435c65447dee2b764623d6c318.zip | |
[NVHPC] add a possibility to control default CUDA version (#38909)
* add a possibility to control default cuda version
* fix stype
* style fix
* resolve comment
* resolve comment
* Fix style in nvhpc package.py
---------
Co-authored-by: antonk <antonk@cscs.ch>
Co-authored-by: Mikael Simberg <mikael.simberg@iki.fi>
| -rw-r--r-- | var/spack/repos/builtin/packages/nvhpc/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/nvhpc/package.py b/var/spack/repos/builtin/packages/nvhpc/package.py index cd52b8a634..7579afd873 100644 --- a/var/spack/repos/builtin/packages/nvhpc/package.py +++ b/var/spack/repos/builtin/packages/nvhpc/package.py @@ -355,6 +355,9 @@ class Nvhpc(Package): ) variant("lapack", default=True, description="Enable LAPACK") variant("mpi", default=False, description="Enable MPI") + variant( + "default_cuda", default="default", description="Default CUDA version, for example 11.8" + ) provides("blas", when="+blas") provides("lapack", when="+lapack") @@ -373,6 +376,8 @@ class Nvhpc(Package): env.set("NVHPC_SILENT", "true") env.set("NVHPC_ACCEPT_EULA", "accept") env.set("NVHPC_INSTALL_DIR", self.prefix) + if self.spec.variants["default_cuda"].value != "default": + env.set("NVHPC_DEFAULT_CUDA", self.spec.variants["default_cuda"].value) if self.spec.variants["install_type"].value == "network": local_dir = join_path(self._version_prefix(), "share_objects") |