Add support for compiling CUDA code on both amd64 and the orin
Change-Id: Ic7aca062c454fa1da3480667363e7250ddc2a8fe
Signed-off-by: Austin Schuh <austin.linux@gmail.com>
diff --git a/third_party/BUILD b/third_party/BUILD
index 7195123..0430621 100644
--- a/third_party/BUILD
+++ b/third_party/BUILD
@@ -117,3 +117,23 @@
"//conditions:default": [":unavailable"],
}),
)
+
+cc_library(
+ name = "cudart",
+ visibility = ["//visibility:public"],
+ deps = select({
+ "//tools:cpu_k8": ["@amd64_debian_sysroot//:cudart"],
+ "//tools:cpu_arm64": ["@arm64_debian_sysroot//:cudart"],
+ "//conditions:default": [":unavailable"],
+ }),
+)
+
+cc_library(
+ name = "nppi",
+ visibility = ["//visibility:public"],
+ deps = select({
+ "//tools:cpu_k8": ["@amd64_debian_sysroot//:nppi"],
+ "//tools:cpu_arm64": ["@arm64_debian_sysroot//:nppi"],
+ "//conditions:default": [":unavailable"],
+ }),
+)