commit ae856ca9c0df9d5485db46bbd341322e652c15e4
author Austin Schuh <austin.linux@gmail.com> Sat Nov 18 14:04:00 2023 -0800
committer Austin Schuh <austin.linux@gmail.com> Sat Nov 18 17:11:22 2023 -0800
tree a598d77e10d81cdcd2120c4cf6d199da81473bc1
parent 0bd410a4d1e2f8c11eebcdde03827adde8edd3e4

Add support for compiling CUDA code on both amd64 and the orin

Change-Id: Ic7aca062c454fa1da3480667363e7250ddc2a8fe
Signed-off-by: Austin Schuh <austin.linux@gmail.com>

WORKSPACE

diff --git a/WORKSPACE b/WORKSPACE
index efeb2a4..e60524a 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -1600,3 +1600,23 @@
     strip_prefix = "julia-1.8.5",
     url = "https://julialang-s3.julialang.org/bin/linux/x64/1.8/julia-1.8.5-linux-x86_64.tar.gz",
 )
+
+http_archive(
+    name = "com_github_nvidia_cuco",
+    build_file = "//third_party/cuco:cuco.BUILD",
+    patch_args = ["-p1"],
+    patches = [
+        "//third_party/cuco:template.patch",
+    ],
+    sha256 = "eecc9a111956a195f28ebc4b4fd23ac6991d072f5c1d7c68a59d059e05d7ad78",
+    strip_prefix = "cuCollections-b7514d2010967fdfe4a1d414894bb945bc09fddc",
+    url = "https://github.com/NVIDIA/cuCollections/archive/b7514d2010967fdfe4a1d414894bb945bc09fddc.zip",
+)
+
+http_archive(
+    name = "com_github_nvidia_cccl",
+    build_file = "//third_party/cccl:cccl.BUILD",
+    sha256 = "38160c628a9e32b7cd55553f299768f72b24074cc9c1a993ba40a177877b3421",
+    strip_prefix = "cccl-931dc6793482c61edbc97b7a19256874fd264313",
+    url = "https://github.com/NVIDIA/cccl/archive/931dc6793482c61edbc97b7a19256874fd264313.zip",
+)