py-tensorflow: fix numa build (#44607)

2024-07-03 08:34:34 +02:00 · 2024-07-03 08:34:34 +02:00 · 9dd92f493a
commit 9dd92f493a
parent b23e832002
1 changed files with 6 additions and 2 deletions
--- a/var/spack/repos/builtin/packages/py-tensorflow/package.py
+++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py
@ -799,8 +799,12 @@ def build(self, spec, prefix):
        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")