aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-26 11:16:17 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-26 11:19:40 -0700
commitd778904d942dcfcda8d21fe990cbb4e20102cfe9 (patch)
tree1162767de64709d908306b03f6c7fb4236447cac
parenta4476d5bfefb6605a42d77da4598d5a8125e638d (diff)
Fixes to cuda_config.h.
PiperOrigin-RevId: 160168545
-rw-r--r--third_party/toolchains/gpus/cuda/cuda/cuda_config.h13
1 files changed, 10 insertions, 3 deletions
diff --git a/third_party/toolchains/gpus/cuda/cuda/cuda_config.h b/third_party/toolchains/gpus/cuda/cuda/cuda_config.h
index dddee32193..f6662274cc 100644
--- a/third_party/toolchains/gpus/cuda/cuda/cuda_config.h
+++ b/third_party/toolchains/gpus/cuda/cuda/cuda_config.h
@@ -14,7 +14,14 @@ limitations under the License.
==============================================================================*/
// DO NOT EDIT: automatically generated file
-#ifndef THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_
-#define THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_
+#ifndef CUDA_CUDA_CONFIG_H_
+#define CUDA_CUDA_CONFIG_H_
-#endif // THIRD_PARTY_TENSORFLOW_OPENSOURCE_ONLY_TOOLCHAINS_GPUS_CUDA_CUDA_CONFIG_H_
+#define TF_CUDA_CAPABILITIES CudaVersion("3.0")
+
+#define TF_CUDA_VERSION "8.0"
+#define TF_CUDNN_VERSION "5"
+
+#define TF_CUDA_TOOLKIT_PATH "/usr/local/cuda-8.0"
+
+#endif // CUDA_CUDA_CONFIG_H_