From 5d7d5a76a339a7ee947aa2e7df812787346e01a5 Mon Sep 17 00:00:00 2001
From: "Wen-Heng (Jack) Chung" <whchung@gmail.com>
Date: Tue, 15 Jan 2019 22:18:50 +0000
Subject: [PATCH] [CUDA] Fix invalid nvcc configuration

---
 third_party/gpus/cuda_configure.bzl | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl
index af8876ba175..f8f3e907084 100644
--- a/third_party/gpus/cuda_configure.bzl
+++ b/third_party/gpus/cuda_configure.bzl
@@ -1409,9 +1409,11 @@ def _create_local_cuda_repository(repository_ctx):
     # bazel's header check failing.
     cuda_defines["%{extra_no_canonical_prefixes_flags}"] = (
         "flag: \"-fno-canonical-system-headers\"")
-    nvcc_path = "cuda/bin/nvcc"
-    if _is_windows(repository_ctx):
-      nvcc_path += ".exe"
+    nvcc_path = str(
+        repository_ctx.path("%s/bin/nvcc%s" % (
+            cuda_config.cuda_toolkit_path,
+            ".exe" if _is_windows(repository_ctx) else "",
+        )))
     _tpl(
         repository_ctx,
         "crosstool:BUILD",