zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add cuda as a variant to opa-psm2 (#39754)

Browse Source 
Signed-off-by: Raafat Feki <rfeki@cornelisnetworks.com>
This commit is contained in:
rfeki
2023-09-18 14:28:37 -05:00
committed by GitHub
parent d111bde69e
commit 60ce6c7302
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/opa-psm2/package.py
View File

@@ -6,7 +6,7 @@
from spack.package import * from spack.package import *
class OpaPsm2(MakefilePackage): class OpaPsm2(MakefilePackage, CudaPackage):
"""Omni-Path Performance Scaled Messaging 2 (PSM2) library""" """Omni-Path Performance Scaled Messaging 2 (PSM2) library"""
homepage = "https://github.com/cornelisnetworks/opa-psm2" homepage = "https://github.com/cornelisnetworks/opa-psm2"
@@ -31,6 +31,7 @@ class OpaPsm2(MakefilePackage):
variant("avx2", default=True, description="Enable AVX2 instructions") variant("avx2", default=True, description="Enable AVX2 instructions")
depends_on("numactl") depends_on("numactl")
depends_on("cuda@8:", when="+cuda")
# patch to get the Makefile to use the spack compiler wrappers # patch to get the Makefile to use the spack compiler wrappers
patch( patch(
@@ -45,6 +46,8 @@ def setup_build_environment(self, env):
# this variable must be set when we use the Intel compilers to # this variable must be set when we use the Intel compilers to
# ensure that the proper flags are set # ensure that the proper flags are set
env.set("CCARCH", "icc") env.set("CCARCH", "icc")
if "+cuda" in self.spec:
env.set("PSM_CUDA", "1")
def edit(self, spec, prefix): def edit(self, spec, prefix):
# Change the makefile so libraries and includes are not # Change the makefile so libraries and includes are not