experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove no longer used compute capability list macro in C++ header.

Browse Source 
PiperOrigin-RevId: 308005305
Change-Id: I71f1f0ab3b1fd33958bf28c563b0fdc3c68db5a4
This commit is contained in:
Christian Sigg 2020-04-23 02:06:04 -07:00 committed by TensorFlower Gardener
parent 8643ea57a0
commit 1588f45ee5
3 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/common_runtime/gpu/gpu_device.cc
View File

@ -1444,7 +1444,7 @@ struct CudaVersion {
}; };
std::vector<CudaVersion> supported_cuda_compute_capabilities = { std::vector<CudaVersion> supported_cuda_compute_capabilities = {
TF_CUDA_CAPABILITIES,}; CudaVersion("3.5"), CudaVersion("5.2")};
std::vector<CudaVersion> GetSupportedCudaComputeCapabilities() { std::vector<CudaVersion> GetSupportedCudaComputeCapabilities() {
auto cuda_caps = supported_cuda_compute_capabilities; auto cuda_caps = supported_cuda_compute_capabilities;

2
third_party/gpus/cuda/cuda_config.h.tpl vendored
View File

@ -16,8 +16,6 @@ limitations under the License.
#ifndef CUDA_CUDA_CONFIG_H_ #ifndef CUDA_CUDA_CONFIG_H_
#define CUDA_CUDA_CONFIG_H_ #define CUDA_CUDA_CONFIG_H_
#define TF_CUDA_CAPABILITIES %{cuda_compute_capabilities}
#define TF_CUDA_VERSION "%{cuda_version}" #define TF_CUDA_VERSION "%{cuda_version}"
#define TF_CUDA_LIB_VERSION "%{cuda_lib_version}" #define TF_CUDA_LIB_VERSION "%{cuda_lib_version}"
#define TF_CUDNN_VERSION "%{cudnn_version}" #define TF_CUDNN_VERSION "%{cudnn_version}"

8
third_party/gpus/cuda_configure.bzl vendored
View File

@ -771,10 +771,6 @@ filegroup(name="cudnn-include")
"%{cuda_version}": "", "%{cuda_version}": "",
"%{cuda_lib_version}": "", "%{cuda_lib_version}": "",
"%{cudnn_version}": "", "%{cudnn_version}": "",
"%{cuda_compute_capabilities}": ",".join([
"CudaVersion(\"%s\")" % c
for c in _DEFAULT_CUDA_COMPUTE_CAPABILITIES
]),
"%{cuda_toolkit_path}": "", "%{cuda_toolkit_path}": "",
}, },
"cuda/cuda/cuda_config.h", "cuda/cuda/cuda_config.h",
@ -1140,10 +1136,6 @@ def _create_local_cuda_repository(repository_ctx):
"%{cuda_version}": cuda_config.cuda_version, "%{cuda_version}": cuda_config.cuda_version,
"%{cuda_lib_version}": cuda_config.cuda_lib_version, "%{cuda_lib_version}": cuda_config.cuda_lib_version,
"%{cudnn_version}": cuda_config.cudnn_version, "%{cudnn_version}": cuda_config.cudnn_version,
"%{cuda_compute_capabilities}": ", ".join([
"CudaVersion(\"%s\")" % c
for c in cuda_config.compute_capabilities
]),
"%{cuda_toolkit_path}": cuda_config.cuda_toolkit_path, "%{cuda_toolkit_path}": cuda_config.cuda_toolkit_path,
}, },
) )