From a4e303ef3f48510b2e41b6d307e3ddaac43eb606 Mon Sep 17 00:00:00 2001
From: jnnnnn <jnnnnn@gmail.com>
Date: Mon, 29 May 2023 19:17:48 +1200
Subject: [PATCH] cfg_if

---
 sys/build.rs | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/sys/build.rs b/sys/build.rs
index 02461a5..170a1d4 100644
--- a/sys/build.rs
+++ b/sys/build.rs
@@ -36,16 +36,15 @@ fn main() {
         println!("cargo:rustc-link-lib=cublas");
         println!("cargo:rustc-link-lib=cudart");
         println!("cargo:rustc-link-lib=cublasLt");
-        #[cfg(target_os = "windows")]
-        {
-            let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");
-            println!("cargo:rustc-link-search={}", cuda_path.display());
-        }
-        #[cfg(not(target_os = "windows"))]
-        {
-            println!("cargo:rustc-link-lib=culibos");
-            println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
-            println!("cargo:rustc-link-search=/opt/cuda/lib64");
+        cfg_if! {
+            if #[cfg(target_os = "windows")] {
+                let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");
+                println!("cargo:rustc-link-search={}", cuda_path.display());
+            } else {
+                println!("cargo:rustc-link-lib=culibos");
+                println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
+                println!("cargo:rustc-link-search=/opt/cuda/lib64");
+            }
         }
     }
     println!("cargo:rerun-if-changed=wrapper.h");