Symlink files for cuda configuration one level below the BUILD file (i.e., one level under local_config_cuda/cuda).

PiperOrigin-RevId: 159570549

1 files changed, 55 insertions, 27 deletions

diff --git a/third_party/gpus/cuda/BUILD.tpl b/third_party/gpus/cuda/BUILD.tpl
index f7610dd7a9..2cca438d6d 100644
--- a/third_party/gpus/cuda/BUILD.tpl
+++ b/third_party/gpus/cuda/BUILD.tpl
@@ -40,20 +40,23 @@ config_setting(
 cc_library(
     name = "cuda_headers",
     hdrs = [
-        "cuda_config.h",
+        "cuda/cuda_config.h",
         %{cuda_headers}
     ],
     includes = [
         ".",
-        "include",
+        "cuda/include",
     ],
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cudart_static",
-    srcs = ["lib/%{cudart_static_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cudart_static_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkopts = select({
         ":freebsd": [],
         "//conditions:default": ["-ldl"],
@@ -66,34 +69,46 @@ cc_library(
 
 cc_library(
     name = "cuda_driver",
-    srcs = ["lib/%{cuda_driver_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cuda_driver_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cudart",
-    srcs = ["lib/%{cudart_lib}"],
-    data = ["lib/%{cudart_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cudart_lib}"],
+    data = ["cuda/lib/%{cudart_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cublas",
-    srcs = ["lib/%{cublas_lib}"],
-    data = ["lib/%{cublas_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cublas_lib}"],
+    data = ["cuda/lib/%{cublas_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cusolver",
-    srcs = ["lib/%{cusolver_lib}"],
-    data = ["lib/%{cusolver_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cusolver_lib}"],
+    data = ["cuda/lib/%{cusolver_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     linkopts = ["-lgomp"],
     visibility = ["//visibility:public"],
@@ -101,27 +116,36 @@ cc_library(
 
 cc_library(
     name = "cudnn",
-    srcs = ["lib/%{cudnn_lib}"],
-    data = ["lib/%{cudnn_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cudnn_lib}"],
+    data = ["cuda/lib/%{cudnn_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cufft",
-    srcs = ["lib/%{cufft_lib}"],
-    data = ["lib/%{cufft_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{cufft_lib}"],
+    data = ["cuda/lib/%{cufft_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "curand",
-    srcs = ["lib/%{curand_lib}"],
-    data = ["lib/%{curand_lib}"],
-    includes = ["include"],
+    srcs = ["cuda/lib/%{curand_lib}"],
+    data = ["cuda/lib/%{curand_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     linkstatic = 1,
     visibility = ["//visibility:public"],
 )
@@ -142,19 +166,23 @@ cc_library(
 cc_library(
     name = "cupti_headers",
     hdrs = [
-        "cuda_config.h",
+        "cuda/cuda_config.h",
         ":cuda-extras",
     ],
     includes = [
         ".",
-        "extras/CUPTI/include/",
+        "cuda/extras/CUPTI/include/",
     ],
     visibility = ["//visibility:public"],
 )
 
 cc_library(
     name = "cupti_dsos",
-    data = ["lib/%{cupti_lib}"],
+    data = ["cuda/lib/%{cupti_lib}"],
+    includes = [
+        ".",
+        "cuda/include",
+    ],
     visibility = ["//visibility:public"],
 )