[tvm] 03/23: Update build.rs to use new tvm-build version

[jr...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jroesch pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit 1939a4d5eb0965b9f2c11e042dc2dcb00797f1f5
Author: Jared Roesch <jr...@octoml.ai>
AuthorDate: Sat Feb 20 01:09:58 2021 -0800

    Update build.rs to use new tvm-build version
---
 rust/tvm-sys/build.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/rust/tvm-sys/build.rs b/rust/tvm-sys/build.rs
index 0638f71..78ff84e 100644
--- a/rust/tvm-sys/build.rs
+++ b/rust/tvm-sys/build.rs
@@ -58,8 +58,8 @@ fn find_using_tvm_build() -> Result<TVMInstall> {
     build_config.repository = Some("https://github.com/jroesch/tvm".to_string());
     build_config.branch = Some("rust-tvm-build".to_string());
     let build_result = tvm_build::build(build_config)?;
-    let source_path = build_result.revision_path.join("source");
-    let build_path = build_result.revision_path.join("build");
+    let source_path = build_result.revision.source_path();
+    let build_path = build_result.revision.build_path();
     Ok(TVMInstall {
         source_path,
         build_path,
@@ -76,13 +76,14 @@ fn main() -> Result<()> {
 
     if cfg!(feature = "static-linking") {
         println!("cargo:rustc-link-lib=static=tvm");
-        println!("cargo:rustc-link-search=native={}", build_path.display());
+        // TODO move this to tvm-build as library_path?
+        println!("cargo:rustc-link-search=native={}/build", build_path.display());
 
     }
 
     if cfg!(feature = "dynamic-linking") {
         println!("cargo:rustc-link-lib=dylib=tvm");
-        println!("cargo:rustc-link-search=native={}", build_path.display());
+        println!("cargo:rustc-link-search=native={}/build", build_path.display());
     }
 
     let runtime_api = source_path.join("include/tvm/runtime/c_runtime_api.h");