summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/amgx/package.py2
-rw-r--r--var/spack/repos/builtin/packages/lammps/package.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/amgx/package.py b/var/spack/repos/builtin/packages/amgx/package.py
index ece1d67ee2..c03f3599df 100644
--- a/var/spack/repos/builtin/packages/amgx/package.py
+++ b/var/spack/repos/builtin/packages/amgx/package.py
@@ -42,7 +42,7 @@ class Amgx(CMakePackage, CudaPackage):
if '+cuda' in self.spec:
args.append('-DWITH_CUDA=ON')
cuda_arch = self.spec.variants['cuda_arch'].value
- if cuda_arch is not None:
+ if cuda_arch != 'none':
args.append('-DCUDA_ARCH={0}'.format(cuda_arch[0]))
else:
args.append('-DWITH_CUDA=OFF')
diff --git a/var/spack/repos/builtin/packages/lammps/package.py b/var/spack/repos/builtin/packages/lammps/package.py
index 2320cf0050..5bf2986858 100644
--- a/var/spack/repos/builtin/packages/lammps/package.py
+++ b/var/spack/repos/builtin/packages/lammps/package.py
@@ -150,7 +150,7 @@ class Lammps(CMakePackage, CudaPackage):
args.append('-DPKG_GPU=ON')
args.append('-DGPU_API=cuda')
cuda_arch = spec.variants['cuda_arch'].value
- if cuda_arch is not None:
+ if cuda_arch != 'none':
args.append('-DGPU_ARCH=sm_{0}'.format(cuda_arch[0]))
args.append('-DCUDA_MPS_SUPPORT={0}'.format(
'ON' if '+cuda_mps' in spec else 'OFF'))