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

Update tensorflow variant defaults to match upstream defaults (#37610)

Browse Source 
* Update tensorflow variant defaults to match project's defaults

* Apply code style
This commit is contained in:
Lehman Garrison 2023-05-12 11:27:51 -04:00 committed by GitHub
parent fe8734cd52
commit 9b35c3cdcc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -129,13 +129,15 @@ class PyTensorflow(Package, CudaPackage, ROCmPackage, PythonExtension):
variant("aws", default=False, description="Build with Amazon AWS Platform support") variant("aws", default=False, description="Build with Amazon AWS Platform support")
variant("kafka", default=False, description="Build with Apache Kafka Platform support") variant("kafka", default=False, description="Build with Apache Kafka Platform support")
variant("ignite", default=False, description="Build with Apache Ignite 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("gdr", default=False, description="Build with GDR support")
variant("verbs", default=False, description="Build with libverbs support") variant("verbs", default=False, description="Build with libverbs support")
variant("ngraph", default=False, description="Build with Intel nGraph support") variant("ngraph", default=False, description="Build with Intel nGraph support")
variant("opencl", default=False, description="Build with OpenCL SYCL support") variant("opencl", default=False, description="Build with OpenCL SYCL support")
variant("computecpp", default=False, description="Build with ComputeCPP 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("cuda", default=sys.platform != "darwin", description="Build with CUDA support")
variant( variant(
"nccl", default=sys.platform.startswith("linux"), description="Enable NVIDIA NCCL support" "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("monolithic", default=False, description="Static monolithic build")
variant("numa", default=False, description="Build with NUMA support") variant("numa", default=False, description="Build with NUMA support")
variant( 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") extends("python")