Skip to content

Commit

Permalink
Address some comments
Browse files Browse the repository at this point in the history
Signed-off-by: leonwanghui <[email protected]>
  • Loading branch information
leonwanghui committed Jul 16, 2020
1 parent a7958bf commit 5696d19
Show file tree
Hide file tree
Showing 3 changed files with 10 additions and 10 deletions.
6 changes: 3 additions & 3 deletions apps/wasm-standalone/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ This project should be considered **experimental** at the very early stage, all
- Download model
```
cd wasm-graphcompiler/tools && wget https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet50v1/resnet50v1.onnx
cd wasm-graph/tools && wget https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet50v1/resnet50v1.onnx
```
- Compile
Expand Down Expand Up @@ -155,8 +155,8 @@ Check the usage of `wasm-graphruntime`:
Usage: wasm-graphruntime [options]

Options:
-c, --ms-backend-config FILE_PATH
set wasm backend config file
-g, --wasm-graph-file FILE_PATH
set the path to wasm graph file
-i, --input-data-file FILE_PATH
set the path to input image file
-l, --label-class-file FILE_PATH
Expand Down
4 changes: 2 additions & 2 deletions apps/wasm-standalone/wasm-graph/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,6 @@ serde = "1.0.53"
serde_derive = "1.0.53"
serde_json = "1.0.53"
ndarray = "0.12"
tvm-common = { version = "0.1", path = "../tvm/rust/common" }
tvm-runtime = { version = "0.1", path = "../tvm/rust/runtime" }
tvm-common = { version = "0.1", path = "../../../rust/common" }
tvm-runtime = { version = "0.1", path = "../../../rust/runtime" }
lazy_static = "1.1.1"
10 changes: 5 additions & 5 deletions apps/wasm-standalone/wasm-graphruntime/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,9 @@ fn main() {

let mut opts = Options::new();
opts.optopt(
"c",
"ms-backend-config",
"set wasm backend config file",
"g",
"wasm-graph-file",
"set the path to wasm graph file",
"FILE_PATH",
);
opts.optopt(
Expand All @@ -69,7 +69,7 @@ fn main() {
print_usage(&program, opts);
return;
}
let wasm_backend_file: String = match matches.opt_str("c") {
let wasm_graph_file: String = match matches.opt_str("g") {
Some(s) => s,
None => String::from(""),
};
Expand All @@ -84,7 +84,7 @@ fn main() {
let img = image::open(input_data_file).unwrap();
let input = data_preprocess(img);

let output: Tensor = match runtime::execute(wasm_backend_file, input) {
let output: Tensor = match runtime::execute(wasm_graph_file, input) {
Ok(m) => m,
Err(f) => panic!(f.to_string()),
};
Expand Down

0 comments on commit 5696d19

Please sign in to comment.