From 10c005554d9be2275be7579f00d572308c4dc9a9 Mon Sep 17 00:00:00 2001 From: Christian Sigg Date: Wed, 3 Jun 2020 02:42:31 -0700 Subject: [PATCH] Change RBE's default compute capability to compute_35,compute_60 (from compute_30,compute_60), because compute_30 is deprecated and will be removed in CUDA 11. We don't build releases for compute_30 and XLA requires compute_35. PiperOrigin-RevId: 314495728 Change-Id: Idf5902adb8e12073bb372592a8da6d9b2289f950 --- third_party/toolchains/preconfig/generate/generate.bzl | 2 +- third_party/toolchains/remote_config/rbe_config.bzl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/third_party/toolchains/preconfig/generate/generate.bzl b/third_party/toolchains/preconfig/generate/generate.bzl index 1c8a4dfb052..95e46c73c51 100644 --- a/third_party/toolchains/preconfig/generate/generate.bzl +++ b/third_party/toolchains/preconfig/generate/generate.bzl @@ -41,7 +41,7 @@ def _tensorflow_rbe_config(name, compiler, python_version, os, rocm_version = No env.update({ "TF_NEED_CUDA": "1", "TF_CUDA_CLANG": "1" if compiler.endswith("clang") else "0", - "TF_CUDA_COMPUTE_CAPABILITIES": "3.0,6.0", + "TF_CUDA_COMPUTE_CAPABILITIES": "3.5,6.0", "TF_ENABLE_XLA": "1", "TF_CUDNN_VERSION": cudnn_version, "TF_CUDA_VERSION": cuda_version, diff --git a/third_party/toolchains/remote_config/rbe_config.bzl b/third_party/toolchains/remote_config/rbe_config.bzl index 744496e8335..08c115ab3af 100644 --- a/third_party/toolchains/remote_config/rbe_config.bzl +++ b/third_party/toolchains/remote_config/rbe_config.bzl @@ -37,7 +37,7 @@ def _tensorflow_rbe_config(name, compiler, python_versions, os, rocm_version = N env.update({ "TF_NEED_CUDA": "1", "TF_CUDA_CLANG": "1" if compiler.endswith("clang") else "0", - "TF_CUDA_COMPUTE_CAPABILITIES": "3.0,6.0", + "TF_CUDA_COMPUTE_CAPABILITIES": "3.5,6.0", "TF_ENABLE_XLA": "1", "TF_CUDNN_VERSION": cudnn_version, "TF_CUDA_VERSION": cuda_version,