diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/opa-psm2/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/opa-psm2/package.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/opa-psm2/package.py b/var/spack/repos/builtin/packages/opa-psm2/package.py index 9aab415750..38d1ab5e97 100644 --- a/var/spack/repos/builtin/packages/opa-psm2/package.py +++ b/var/spack/repos/builtin/packages/opa-psm2/package.py @@ -6,7 +6,7 @@ from spack.package import * -class OpaPsm2(MakefilePackage): +class OpaPsm2(MakefilePackage, CudaPackage): """Omni-Path Performance Scaled Messaging 2 (PSM2) library""" homepage = "https://github.com/cornelisnetworks/opa-psm2" @@ -31,6 +31,7 @@ class OpaPsm2(MakefilePackage): variant("avx2", default=True, description="Enable AVX2 instructions") depends_on("numactl") + depends_on("cuda@8:", when="+cuda") # patch to get the Makefile to use the spack compiler wrappers patch( @@ -45,6 +46,8 @@ class OpaPsm2(MakefilePackage): # this variable must be set when we use the Intel compilers to # ensure that the proper flags are set env.set("CCARCH", "icc") + if "+cuda" in self.spec: + env.set("PSM_CUDA", "1") def edit(self, spec, prefix): # Change the makefile so libraries and includes are not |