fix(sys): link DX12/DML libs when directml feature is enabled

pykeio · Nov 27, 2023 · 6938da1 · 6938da1
1 parent 0777acb
commit 6938da1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ort-sys/build.rs b/ort-sys/build.rs
@@ -120,6 +120,9 @@ fn static_link_prerequisites() {
 		println!("cargo:rustc-link-lib=framework=Foundation");
 	} else if target_os == "linux" || target_os == "android" {
 		println!("cargo:rustc-link-lib=stdc++");
+	} else if target_os == "windows" && cfg!(feature = "directml") {
+		println!("cargo:rustc-link-lib=dxguid");
+		println!("cargo:rustc-link-lib=DirectML");
 	}
 }