From 9dd92f493a8554344f009382e14f6ae751634908 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Wed, 3 Jul 2024 08:34:34 +0200 Subject: py-tensorflow: fix numa build (#44607) --- var/spack/repos/builtin/packages/py-tensorflow/package.py | 8 ++++++-- 1 file 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") -- cgit v1.2.3-70-g09d2