diff options
author | Greg Becker <becker33@llnl.gov> | 2022-11-16 15:01:37 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-16 13:01:37 -0800 |
commit | 1263b5c4449200facdbfe8266267b51ce6fcf97e (patch) | |
tree | 85d3c1651172d7af091d2437766e5b461903e392 | |
parent | 90f0a8eacccd84499a8a8a15eff1ab2ece2a7a6a (diff) | |
download | spack-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.py | 4 | ||||
-rw-r--r-- | var/spack/repos/builtin/packages/aluminum/package.py | 5 |
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", ) |