diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2024-07-03 08:34:34 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-07-03 08:34:34 +0200 |
| commit | 9dd92f493a8554344f009382e14f6ae751634908 (patch) | |
| tree | f01c69e4447969a0884a6ff239f0f99072380fff | |
| parent | b23e8320029411512daa6633f2e12584030f22c6 (diff) | |
| download | spack-9dd92f493a8554344f009382e14f6ae751634908.tar.gz spack-9dd92f493a8554344f009382e14f6ae751634908.tar.bz2 spack-9dd92f493a8554344f009382e14f6ae751634908.tar.xz spack-9dd92f493a8554344f009382e14f6ae751634908.zip | |
py-tensorflow: fix numa build (#44607)
| -rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index f9d531a8e2..4587b04707 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -799,8 +799,12 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): if "~nccl" in spec: args.append("--config=nonccl") - if "+numa" in spec: - args.append("--config=numa") + # https://github.com/tensorflow/tensorflow/issues/63080 + 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") |