Skip to content

Commit 0f95a1b

Browse files
authored
Update README.md
1 parent 3a24ebc commit 0f95a1b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,19 +85,19 @@ We use ```graph_net/benchmark_demo.sh``` to benchmark GraphNet computation graph
8585
bash graph_net/benchmark_demo.sh &
8686
```
8787

88-
The script will run ```graph_net.torch.test_compiler``` with specific batch and log configurations.
88+
The script runs ```graph_net.torch.test_compiler``` with specific batch and log configurations.
8989

9090
Or you can customize and use ```graph_net.torch.test_compiler``` yourself:
9191

9292
```
93-
python3 -m graph_net.torch.test_compiler \
93+
python -m graph_net.torch.test_compiler \
9494
--model-path $GRAPH_NET_EXTRACT_WORKSPACE/model_name/ \
9595
--compiler /path/to/custom/compiler/ \
9696
--output-dir /path/to/save/JSON/result/file/
9797
# Note: if --compiler is omitted, PyTorch’s built-in compiler is used by default
9898
```
9999

100-
After that, ```graph_net.torch.test_compiler``` processes as below:
100+
After that, ```graph_net.torch.test_compiler``` will process as below:
101101
1. Running the original model in eager mode to record a baseline.
102102
2. Compiling the model with the specified backend (e.g., CINN, TorchInductor, TVM).
103103
3. Executing the compiled model and collecting its runtime and outputs.
@@ -108,7 +108,7 @@ After that, ```graph_net.torch.test_compiler``` processes as below:
108108
After processing, we provide ```graph_net/analysis.py``` to generate [violin plot](https://en.m.wikipedia.org/wiki/Violin_plot) based on the JSON results.
109109

110110
```
111-
python3 graph_net/analysis.py \
111+
python -m graph_net.analysis \
112112
--benchmark-path /path/to/read/JSON/result/file/ \
113113
--output-dir /path/to/save/output/figures/
114114
```

0 commit comments

Comments
 (0)