Fix: add optix finding logic for linux and move nvvm logic to helper

Author: RDambrosio016

crates/find_cuda_helper/src/lib.rs

@@ -127,3 +127,32 @@ pub fn find_optix_root() -> Option<PathBuf> {
         .or_else(|| env::var("OPTIX_ROOT_DIR").ok())
         .map(PathBuf::from)
 }
+
+#[cfg(target_family = "unix")]
+pub fn find_optix_root() -> Option<PathBuf> {
+    env::var("OPTIX_ROOT")
+        .ok()
+        .or_else(|| env::var("OPTIX_ROOT_DIR").ok())
+        .map(PathBuf::from)
+}
+
+#[cfg(target_os = "windows")]
+pub fn find_libnvvm_bin_dir() -> String {
+    find_cuda_root()
+        .expect("Failed to find CUDA ROOT, make sure the CUDA SDK is installed and CUDA_PATH or CUDA_ROOT are set!")
+        .join("nvvm")
+        .join("lib")
+        .join("x64")
+        .to_string_lossy()
+        .into_owned()
+}
+
+#[cfg(target_os = "linux")]
+pub fn find_libnvvm_bin_dir() -> String {
+    find_cuda_root()
+        .expect("Failed to find CUDA ROOT, make sure the CUDA SDK is installed and CUDA_PATH or CUDA_ROOT are set!")
+        .join("nvvm")
+        .join("lib64")
+        .to_string_lossy()
+        .into_owned()
+}

crates/nvvm/build.rs

@@ -1,26 +1,6 @@
-use find_cuda_helper::find_cuda_root;
-
-#[cfg(target_os = "windows")]
-fn lib_search_path() -> String {
-    find_cuda_root()
-        .expect("Failed to find CUDA ROOT, make sure the CUDA SDK is installed and CUDA_PATH or CUDA_ROOT are set!")
-        .join("nvvm")
-        .join("lib")
-        .join("x64")
-        .to_string_lossy()
-        .into_owned()
-}
-
-#[cfg(target_os = "linux")]
-fn lib_search_path() -> String {
-    format!("{}/nvvm/lib64", std::env::var("CUDA_ROOT").unwrap())
-}
-
-fn libnvvm_build() {
-    println!("cargo:rustc-link-search={}", lib_search_path());
-    println!("cargo:rustc-link-lib=dylib=nvvm");
-}
+use find_cuda_helper::find_libnvvm_bin_dir;
 
 fn main() {
-    libnvvm_build()
+    println!("cargo:rustc-link-search={}", find_libnvvm_bin_dir());
+    println!("cargo:rustc-link-lib=dylib=nvvm");
 }