1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
use std::error::Error;
use vergen_git2::*;
fn vergen() -> Result<(), Box<dyn Error>> {
// NOTE: This will output everything, and requires all features enabled.
// NOTE: See the specific builder documentation for configuration options.
let build = BuildBuilder::all_build()?;
let git2 = Git2Builder::all_git()?;
let cargo = CargoBuilder::all_cargo()?;
let status = Emitter::default()
.add_instructions(&build)?
.add_instructions(&git2)?
.add_instructions(&cargo)?
.fail_on_error()
.emit();
// set the env variable myself if the status failed
if status.is_err() {
eprintln!("Failed to get version information from git");
eprintln!("Likely building a published version from cargo or bioconda.");
std::env::set_var("VERGEN_GIT_DESCRIBE", "unknown");
}
Ok(())
}
fn main() {
/*
use std::process::Command;
let output = Command::new("git")
.args(["rev-parse", "HEAD"])
.output()
.unwrap();
let git_hash = String::from_utf8(output.stdout).unwrap();
let version = env!("CARGO_PKG_VERSION");
println!(
"cargo:rustc-env=CARGO_GIT_HASH={}
cargo:rustc-env=CARGO_LONG_VERSION={} commit:{}",
git_hash, version, git_hash
);
*/
vergen().expect("Unable to set version and git hash");
// Generate the model code and state file from the ONNX file.
use burn_import::onnx::ModelGen;
use burn_import::onnx::RecordType;
for x in &[
"src/m6a_burn/two_zero.onnx",
"src/m6a_burn/two_two.onnx",
"src/m6a_burn/three_two.onnx",
"src/m6a_burn/revio.onnx",
] {
ModelGen::new()
.input(x) // Path to the ONNX model
.out_dir("m6a_burn/") // Directory for the generated Rust source file (under target/)
.record_type(RecordType::Bincode)
.embed_states(true)
.run_from_script();
}
}