fix: Do not assume libcupti.so location (#253)

bazel-contrib · Jun 18, 2024 · fb21b0e · fb21b0e
1 parent 1fad6ba
commit fb21b0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda/runtime/BUILD.local_cuda b/cuda/runtime/BUILD.local_cuda
@@ -210,7 +210,7 @@ cc_library(
 # CUPTI
 cc_import_versioned_sos(
     name = "cupti_so",
-    shared_library = "cuda/lib64/libcupti.so",
+    shared_library = "cuda/**/lib64/libcupti.so",
 )
 
 cc_import(