Spec.__contains__: traverse only lin/run + direct build

This commit is contained in:
Massimiliano Culpo 2024-11-12 09:20:15 +01:00
parent 27e2e146e2
commit bec6b06c16
No known key found for this signature in database
GPG Key ID: 3E52BB992233066C

View File

@ -97,14 +97,16 @@ class LmodConfiguration(BaseConfiguration):
default_projections = {"all": "{name}/{version}"}
compiler: Optional[spack.spec.Spec]
def __init__(self, spec: spack.spec.Spec, module_set_name: str, explicit: bool) -> None:
super().__init__(spec, module_set_name, explicit)
# FIXME (compiler as nodes): make this a bit more robust
candidates = collections.defaultdict(list)
for node in spec.traverse(deptype=("link", "run")):
candidates["c"].extend(node.dependencies(virtuals="c"))
candidates["cxx"].extend(node.dependencies(virtuals="c"))
candidates["c"].extend(node.dependencies(virtuals=("c",)))
candidates["cxx"].extend(node.dependencies(virtuals=("c",)))
# FIXME (compiler as nodes): decide what to do when we have more than one C compiler
if candidates["c"] and len(set(candidates["c"])) == 1: