diff options
author | Lehman Garrison <lgarrison@flatironinstitute.org> | 2023-05-12 11:27:51 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-12 10:27:51 -0500 |
commit | 9b35c3cdccf30771a06423cb413c4139d8131f16 (patch) | |
tree | e7e565fedb9eaf34de54b2a8e6c9a2496153d583 /var | |
parent | fe8734cd52d161142cccdb87d8fa4e4ae65358e8 (diff) | |
download | spack-9b35c3cdccf30771a06423cb413c4139d8131f16.tar.gz spack-9b35c3cdccf30771a06423cb413c4139d8131f16.tar.bz2 spack-9b35c3cdccf30771a06423cb413c4139d8131f16.tar.xz spack-9b35c3cdccf30771a06423cb413c4139d8131f16.zip |
Update tensorflow variant defaults to match upstream defaults (#37610)
* Update tensorflow variant defaults to match project's defaults
* Apply code style
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/py-tensorflow/package.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-tensorflow/package.py b/var/spack/repos/builtin/packages/py-tensorflow/package.py index 587e656df9..35f33eb919 100644 --- a/var/spack/repos/builtin/packages/py-tensorflow/package.py +++ b/var/spack/repos/builtin/packages/py-tensorflow/package.py @@ -129,13 +129,15 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): variant("aws", default=False, description="Build with Amazon AWS Platform support") variant("kafka", default=False, description="Build with Apache Kafka Platform support") variant("ignite", default=False, description="Build with Apache Ignite support") - variant("xla", default=False, description="Build with XLA JIT support") + variant("xla", default=sys.platform != "darwin", description="Build with XLA JIT support") variant("gdr", default=False, description="Build with GDR support") variant("verbs", default=False, description="Build with libverbs support") variant("ngraph", default=False, description="Build with Intel nGraph support") variant("opencl", default=False, description="Build with OpenCL SYCL support") variant("computecpp", default=False, description="Build with ComputeCPP support") - variant("tensorrt", default=False, description="Build with TensorRT support") + variant( + "tensorrt", default=False, description="Build with TensorRT support" + ) # TODO: enable when TensorRT in Spack variant("cuda", default=sys.platform != "darwin", description="Build with CUDA support") variant( "nccl", default=sys.platform.startswith("linux"), description="Enable NVIDIA NCCL support" @@ -146,7 +148,9 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension): variant("monolithic", default=False, description="Static monolithic build") variant("numa", default=False, description="Build with NUMA support") variant( - "dynamic_kernels", default=False, description="Build kernels into separate shared objects" + "dynamic_kernels", + default=sys.platform.startswith("linux"), + description="Build kernels into separate shared objects", ) extends("python") |