Merge pull request #6 from NarrativeApp/tvm-config-opts

Add user settings to BuildConfig
octoml · Aug 5, 2021 · e385f06 · e385f06
2 parents aff6717 + 3c5323f
commit e385f06
Show file tree

Hide file tree

Showing 3 changed files with 359 additions and 76 deletions.
diff --git a/src/bin/tvm-build.rs b/src/bin/tvm-build.rs
@@ -1,6 +1,6 @@
 use structopt::StructOpt;
 use tracing_subscriber;
-use tvm_build::{self, build, BuildConfig};
+use tvm_build::{self, build, BuildConfig, UserSettings};
 
 #[derive(StructOpt, Debug)]
 #[structopt()]
@@ -13,6 +13,8 @@ struct InstallCommand {
     clean: bool,
     #[structopt(short, long)]
     verbose: bool,
+    #[structopt(flatten)]
+    settings: UserSettings,
 }
 
 #[derive(StructOpt, Debug)]
@@ -49,6 +51,7 @@ fn main() -> anyhow::Result<()> {
             config.clean = install_cmd.clean;
             config.repository = install_cmd.repository;
             config.verbose = install_cmd.verbose;
+            config.settings = install_cmd.settings;
             build(config)?;
             Ok(())
         }