From 1263b5c4449200facdbfe8266267b51ce6fcf97e Mon Sep 17 00:00:00 2001 From: Greg Becker Date: Wed, 16 Nov 2022 15:01:37 -0600 Subject: initial implementation of slingshot detection (#33793) --- lib/spack/spack/platforms/cray.py | 4 ++++ 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", ) -- cgit v1.2.3-60-g2f50