From 60ce6c7302cb0531ce8a6cdd8e0fb60937dfe31a Mon Sep 17 00:00:00 2001 From: rfeki <113144418+rfeki@users.noreply.github.com> Date: Mon, 18 Sep 2023 14:28:37 -0500 Subject: Add cuda as a variant to opa-psm2 (#39754) Signed-off-by: Raafat Feki --- var/spack/repos/builtin/packages/opa-psm2/package.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'var') 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 -- cgit v1.2.3-60-g2f50