zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-tensorflow: fix numa build (#44607)

Browse Source
This commit is contained in:
Adam J. Stewart 2024-07-03 08:34:34 +02:00 committed by GitHub
parent b23e832002
commit 9dd92f493a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/py-tensorflow/package.py
View File

@ -799,8 +799,12 @@ def build(self, spec, prefix):
if "~nccl" in spec: if "~nccl" in spec:
args.append("--config=nonccl") args.append("--config=nonccl")
if "+numa" in spec: # https://github.com/tensorflow/tensorflow/issues/63080
args.append("--config=numa") if self.spec.satisfies("@2.14:"):
args.append(f"--define=with_numa_support={'+numa' in spec}")
else:
if "+numa" in spec:
args.append("--config=numa")
args.append("--config=v2") args.append("--config=v2")