fix(//third_party/cublas): cuBLAS source was incorrect, breaking out

cuBLAS into its own dependency should follow the same workflow as CUDA
(i.e. local only)

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

third_party/cublas/BUILD

@@ -0,0 +1,33 @@
+package(default_visibility = ["//visibility:public"])
+
+config_setting(
+   name = "aarch64",
+   values = {
+        "cpu" : "aarch64",
+   },
+)
+
+cc_library(
+    name = "cublas_headers",
+    hdrs = ["include/cublas.h"] + glob(["include/cublas+.h"]),
+    includes = ["include/"],
+    visibility = ["//visibility:private"],
+)
+
+cc_import(
+    name = "cublas_lib",
+    shared_library = select({
+        "aarch64" : "lib/aarch64-linux-gnu/libcublas.so.10",
+        "//conditions:default" : "lib/x86_64-linux-gnu/libcublas.so.10",
+    }),
+    visibility = ["//visibility:private"],
+)
+
+cc_library(
+    name = "cublas",
+    deps = [
+        "cublas_headers",
+        "cublas_lib"
+    ],
+    visibility = ["//visibility:public"],
+)

third_party/cuda/BUILD

@@ -1,10 +1,22 @@
 package(default_visibility = ["//visibility:public"])
 
+config_setting(
+   name = "aarch64",
+   values = {
+	"cpu" : "aarch64",
+   },
+)
+
 cc_library(
     name = "cudart",
-    srcs = glob([
-        "lib/**/libcudart.so",
-    ]),
+    srcs = select({
+	":aarch64": [
+            "targets/aarch64-linux-gnu/lib/libcudart.so",
+        ],
+	"//conditions:default": [
+	    "targets/x86_64-linux/lib/libcudart.so",
+	]
+    }),
     hdrs = glob([
         "include/**/*.h",
         "include/**/*.hpp",
@@ -15,16 +27,26 @@ cc_library(
 
 cc_library(
     name = "nvToolsExt",
-    srcs = glob([
-        "lib/**/libnvToolsExt.so.1"
-    ])
+    srcs = select({
+        ":aarch64": [
+            "targets/aarch64-linux-gnu/lib/libnvToolsExt.so.1",
+        ],
+        "//conditions:default": [
+            "targets/x86_64-linux/lib/libnvToolsExt.so.1",
+        ]
+    }),
 )
 
 cc_library(
     name = "cuda",
-    srcs = glob([
-        "lib/**/*libcuda.so",
-    ]),
+    srcs = select({
+        ":aarch64": glob([
+            "targets/aarch64-linux-gnu/lib/**/lib*.so",
+        ]),
+        "//conditions:default": glob([
+            "targets/x86_64-linux/lib/**/lib*.so",
+        ])
+    }),
     hdrs = glob([
         "include/**/*.h",
         "include/**/*.hpp",
@@ -34,16 +56,3 @@ cc_library(
     linkopts = ["-Wl,-rpath,lib/"],
 )
 
-cc_library(
-    name = "cublas",
-    srcs = glob([
-        "lib/**/*libcublas.so",
-    ]),
-    hdrs = glob([
-        "include/**/*cublas*.h",
-        "include/**/*.hpp",
-        "include/**/*.inl",
-    ]),
-    includes = ["include/"],
-    linkopts = ["-Wl,-rpath,lib/"],
-)