summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGreg Becker <becker33@llnl.gov>2022-11-16 15:01:37 -0600
committerGitHub <noreply@github.com>2022-11-16 13:01:37 -0800
commit1263b5c4449200facdbfe8266267b51ce6fcf97e (patch)
tree85d3c1651172d7af091d2437766e5b461903e392
parent90f0a8eacccd84499a8a8a15eff1ab2ece2a7a6a (diff)
downloadspack-1263b5c4449200facdbfe8266267b51ce6fcf97e.tar.gz
spack-1263b5c4449200facdbfe8266267b51ce6fcf97e.tar.bz2
spack-1263b5c4449200facdbfe8266267b51ce6fcf97e.tar.xz
spack-1263b5c4449200facdbfe8266267b51ce6fcf97e.zip
initial implementation of slingshot detection (#33793)
-rw-r--r--lib/spack/spack/platforms/cray.py4
-rw-r--r--var/spack/repos/builtin/packages/aluminum/package.py5
2 files changed, 7 insertions, 2 deletions
diff --git a/lib/spack/spack/platforms/cray.py b/lib/spack/spack/platforms/cray.py
index dadf114d90..d266f26284 100644
--- a/lib/spack/spack/platforms/cray.py
+++ b/lib/spack/spack/platforms/cray.py
@@ -36,6 +36,10 @@ _ex_craype_dir = "/opt/cray/pe/cpe"
_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)
diff --git a/var/spack/repos/builtin/packages/aluminum/package.py b/var/spack/repos/builtin/packages/aluminum/package.py
index cad16d0193..3b378d9594 100644
--- a/var/spack/repos/builtin/packages/aluminum/package.py
+++ b/var/spack/repos/builtin/packages/aluminum/package.py
@@ -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",
)