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/tools/platforms/BUILD b/tools/platforms/BUILD
index f485894..599f926 100644
--- a/tools/platforms/BUILD
+++ b/tools/platforms/BUILD
@@ -8,6 +8,7 @@
"//tools/platforms/go:has_support",
"//tools/platforms/rust:has_support",
"//tools/platforms/nodejs:has_support",
+ "//tools/platforms/gpu:nvidia",
],
)
@@ -20,6 +21,7 @@
"//tools/platforms/go:lacks_support",
"//tools/platforms/rust:has_support",
"//tools/platforms/nodejs:lacks_support",
+ "//tools/platforms/gpu:nvidia",
],
)
@@ -37,6 +39,7 @@
# handle that, need to figure out how to do that here or switch linkers.
"//tools/platforms/rust:lacks_support",
"//tools/platforms/nodejs:lacks_support",
+ "//tools/platforms/gpu:none",
],
)
@@ -48,6 +51,7 @@
"//tools/platforms/go:lacks_support",
"//tools/platforms/rust:lacks_support",
"//tools/platforms/nodejs:lacks_support",
+ "//tools/platforms/gpu:none",
],
)
@@ -59,6 +63,7 @@
"//tools/platforms/go:lacks_support",
"//tools/platforms/rust:lacks_support",
"//tools/platforms/nodejs:lacks_support",
+ "//tools/platforms/gpu:none",
],
)
diff --git a/tools/platforms/gpu/BUILD b/tools/platforms/gpu/BUILD
new file mode 100644
index 0000000..9686148
--- /dev/null
+++ b/tools/platforms/gpu/BUILD
@@ -0,0 +1,13 @@
+package(default_visibility = ["//visibility:public"])
+
+constraint_setting(name = "gpu")
+
+constraint_value(
+ name = "nvidia",
+ constraint_setting = ":gpu",
+)
+
+constraint_value(
+ name = "none",
+ constraint_setting = ":gpu",
+)