From 679d41ea669c1a435c65447dee2b764623d6c318 Mon Sep 17 00:00:00 2001 From: Anton Kozhevnikov Date: Thu, 31 Aug 2023 16:16:31 +0200 Subject: [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 Co-authored-by: Mikael Simberg --- var/spack/repos/builtin/packages/nvhpc/package.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'var') 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") -- cgit v1.2.3-70-g09d2