Skip to content

Commit a5a172a

Browse files
committed
Update
1 parent 439be2a commit a5a172a

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

graph_net/benchmark_demo.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,14 @@ for i in /work/GraphNet/samples/torchvision/*/; do
77

88
python -m graph_net.torch.test_compiler \
99
--model-path "${i}" \
10+
--compiler "inductor" \
11+
--warmup 3 \
12+
--trials 10 \
1013
--device cuda \
14+
--output-dir "/work/GraphNet/benchmark_logs"
1115

1216
done > torchvision_cuda.log 2>&1
1317

14-
python3 -m graph_net.analysis --benchmark-log-file /work/GraphNet/torchvision_cuda.log
18+
# python3 -m graph_net.analysis --test-compiler-log-file /work/GraphNet/torchvision_cuda.log
19+
20+
python3 -m graph_net.analysis --benchmark-path /work/GraphNet/benchmark

0 commit comments

Comments
 (0)