Update aws-ofi-nccl to use the hwloc option (#43287)

This commit is contained in:
Auriane R 2024-03-21 09:38:40 +01:00 committed by GitHub
parent 0eb1957999
commit 2303332415
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -26,6 +26,7 @@ class AwsOfiNccl(AutotoolsPackage):
depends_on("cuda")
depends_on("nccl")
depends_on("mpi")
depends_on("hwloc")
depends_on("autoconf", type="build")
depends_on("automake", type="build")
depends_on("libtool", type="build")
@ -52,6 +53,7 @@ def configure_args(self):
"--with-cuda={0}".format(spec["cuda"].prefix),
"--with-nccl={0}".format(spec["nccl"].prefix),
"--with-mpi={0}".format(spec["mpi"].prefix),
"--with-hwloc={0}".format(spec["hwloc"].prefix),
]
)