diff options
| author | Tim Moon <moon13@llnl.gov> | 2021-07-28 09:07:24 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-07-28 16:07:24 +0000 |
| commit | c9327649c0def5f05ba5f6ad13eff64ecf0ee20b (patch) | |
| tree | 3897245f12a14bb128b8f0db81499f6f32c46bc6 /var | |
| parent | 68f696af646734bab1298a9b5c92258eac3baefa (diff) | |
| download | spack-c9327649c0def5f05ba5f6ad13eff64ecf0ee20b.tar.gz spack-c9327649c0def5f05ba5f6ad13eff64ecf0ee20b.tar.bz2 spack-c9327649c0def5f05ba5f6ad13eff64ecf0ee20b.tar.xz spack-c9327649c0def5f05ba5f6ad13eff64ecf0ee20b.zip | |
nvshmem: set env variables instead of appending (#25095)
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/nvshmem/package.py | 42 |
1 files changed, 14 insertions, 28 deletions
diff --git a/var/spack/repos/builtin/packages/nvshmem/package.py b/var/spack/repos/builtin/packages/nvshmem/package.py index f55ce169e1..3e2f619b6a 100644 --- a/var/spack/repos/builtin/packages/nvshmem/package.py +++ b/var/spack/repos/builtin/packages/nvshmem/package.py @@ -41,44 +41,30 @@ class Nvshmem(MakefilePackage, CudaPackage): depends_on('nccl', when='+nccl') def setup_build_environment(self, env): - env.append_flags( - 'CUDA_HOME', self.spec['cuda'].prefix) - env.append_flags( - 'NVSHMEM_PREFIX', self.prefix) + env.set('CUDA_HOME', self.spec['cuda'].prefix) + env.set('NVSHMEM_PREFIX', self.prefix) if '+ucx' in self.spec: - env.append_flags( - 'NVSHMEM_UCX_SUPPORT', '1') - env.append_flags( - 'UCX_HOME', self.spec['ucx'].prefix) + env.set('NVSHMEM_UCX_SUPPORT', '1') + env.set('UCX_HOME', self.spec['ucx'].prefix) if '+gdrcopy' in self.spec: - env.append_flags( - 'NVSHMEM_USE_GDRCOPY', '1') - env.append_flags( - 'GDRCOPY_HOME', self.spec['gdrcopy'].prefix) + env.set('NVSHMEM_USE_GDRCOPY', '1') + env.set('GDRCOPY_HOME', self.spec['gdrcopy'].prefix) if '+nccl' in self.spec: - env.append_flags( - 'NVSHMEM_USE_NCCL', '1') - env.append_flags( - 'NCCL_HOME', self.spec['nccl'].prefix) + env.set('NVSHMEM_USE_NCCL', '1') + env.set('NCCL_HOME', self.spec['nccl'].prefix) if '+mpi' in self.spec: - env.append_flags( - 'NVSHMEM_MPI_SUPPORT', '1') - env.append_flags( - 'MPI_HOME', self.spec['mpi'].prefix) + env.set('NVSHMEM_MPI_SUPPORT', '1') + env.set('MPI_HOME', self.spec['mpi'].prefix) if self.spec.satisfies('^spectrum-mpi') or self.spec.satisfies('^openmpi'): - env.append_flags( - 'NVSHMEM_MPI_IS_OMPI', '1') + env.set('NVSHMEM_MPI_IS_OMPI', '1') else: - env.append_flags( - 'NVSHMEM_MPI_IS_OMPI', '0') + env.set('NVSHMEM_MPI_IS_OMPI', '0') if '+shmem' in self.spec: - env.append_flags( - 'NVSHMEM_SHMEM_SUPPORT', '1') - env.append_flags( - 'SHMEM_HOME', self.spec['mpi'].prefix) + env.set('NVSHMEM_SHMEM_SUPPORT', '1') + env.set('SHMEM_HOME', self.spec['mpi'].prefix) |