diff options
Diffstat (limited to 'tensorflow/core/common_runtime/gpu/gpu_device.cc')
-rw-r--r-- | tensorflow/core/common_runtime/gpu/gpu_device.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/core/common_runtime/gpu/gpu_device.cc b/tensorflow/core/common_runtime/gpu/gpu_device.cc index 0ff64dd939..c08abc5689 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_device.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_device.cc @@ -785,10 +785,8 @@ struct CudaVersion { int minor_part = -1; }; -// "configure" uses the specific name to substitute the following string. -// If you change it, make sure you modify "configure" as well. std::vector<CudaVersion> supported_cuda_compute_capabilities = { - CudaVersion("3.5"), CudaVersion("5.2")}; + TF_CUDA_CAPABILITIES,}; std::vector<CudaVersion> GetSupportedCudaComputeCapabilities() { auto cuda_caps = supported_cuda_compute_capabilities; |