From 3e7bb5f2e8c6ee3b191984454e7d638fdfb42a80 Mon Sep 17 00:00:00 2001
From: Jonathan Bentz <jnbntz@gmail.com>
Date: Fri, 20 Dec 2024 12:52:06 -0600
Subject: [PATCH] updated to fix for Linux

---
 Samples/4_CUDA_Libraries/jitLto/CMakeLists.txt | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/Samples/4_CUDA_Libraries/jitLto/CMakeLists.txt b/Samples/4_CUDA_Libraries/jitLto/CMakeLists.txt
index 978d96de..3f3bec5f 100644
--- a/Samples/4_CUDA_Libraries/jitLto/CMakeLists.txt
+++ b/Samples/4_CUDA_Libraries/jitLto/CMakeLists.txt
@@ -28,7 +28,15 @@ target_include_directories(jitLto PRIVATE
     ${CUDAToolkit_INCLUDE_DIRS}
 )
 
-target_link_libraries(jitLto PRIVATE
-    CUDA::cuda_driver
-    CUDA::nvrtc
-)
+
+
+    target_link_libraries(jitLto PRIVATE
+        CUDA::cuda_driver
+        CUDA::nvrtc
+    )
+
+if(CMAKE_SYSTEM_NAME STREQUAL "Linux")
+    target_link_libraries(jitLto PRIVATE
+        ${CUDAToolkit_LIBRARY_DIR}/libnvJitLink.so
+    )
+endif()