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

initial implementation of slingshot detection (#33793)

Browse Source
This commit is contained in:
Greg Becker 2022-11-16 15:01:37 -06:00 committed by GitHub
parent 90f0a8eacc
commit 1263b5c444
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/spack/spack/platforms/cray.py
View File

@ -36,6 +36,10 @@
_xc_craype_dir = "/opt/cray/pe/cdt"
def slingshot_network():
return os.path.exists("/lib64/libcxi.so")
def _target_name_from_craype_target_name(name):
return _craype_name_to_target_name.get(name, name)

5
var/spack/repos/builtin/packages/aluminum/package.py
View File

@ -5,6 +5,7 @@
import os
import spack.platforms.cray
from spack.package import *
@ -53,13 +54,13 @@ class Aluminum(CMakePackage, CudaPackage, ROCmPackage):
variant("rccl", default=False, description="Builds with support for RCCL communication lib")
variant(
"ofi_libfabric_plugin",
default=True,
default=spack.platforms.cray.slingshot_network(),
when="+rccl",
description="Builds with support for OFI libfabric enhanced RCCL/NCCL communication lib",
)
variant(
"ofi_libfabric_plugin",
default=True,
default=spack.platforms.cray.slingshot_network(),
when="+nccl",
description="Builds with support for OFI libfabric enhanced RCCL/NCCL communication lib",
)