We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 439be2a commit a5a172aCopy full SHA for a5a172a
graph_net/benchmark_demo.sh
@@ -7,8 +7,14 @@ for i in /work/GraphNet/samples/torchvision/*/; do
7
8
python -m graph_net.torch.test_compiler \
9
--model-path "${i}" \
10
+ --compiler "inductor" \
11
+ --warmup 3 \
12
+ --trials 10 \
13
--device cuda \
14
+ --output-dir "/work/GraphNet/benchmark_logs"
15
16
done > torchvision_cuda.log 2>&1
17
-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