Use the host libcuda and libcudart
The driver and userspace versions need to match identically, even if we
link against something different. We are better off using the host
version.
This makes it so that upgrading Debian, or using Ubuntu, actually can
run CUDA code.
Change-Id: I234f95b8dafa7421643381d6d4c0c905a58a1cc9
Signed-off-by: Austin Schuh <austin.linux@gmail.com>
diff --git a/compilers/amd64_debian_rootfs.BUILD b/compilers/amd64_debian_rootfs.BUILD
index 57246ae..38e2622 100644
--- a/compilers/amd64_debian_rootfs.BUILD
+++ b/compilers/amd64_debian_rootfs.BUILD
@@ -34,10 +34,14 @@
cc_library(
name = "cudart",
- srcs = [
+ additional_linker_inputs = [
"usr/lib/x86_64-linux-gnu/libcuda.so.1",
"usr/lib/x86_64-linux-gnu/libcudart.so.11.0",
],
+ linkopts = [
+ "external/amd64_debian_sysroot/usr/lib/x86_64-linux-gnu/libcuda.so.1",
+ "external/amd64_debian_sysroot/usr/lib/x86_64-linux-gnu/libcudart.so.11.0",
+ ],
visibility = ["//visibility:public"],
)