Merge pull request #879 from NVIDIA/anuragd/benchmark_reorg

refactor: Restructure of benchmark files to tools
pytorch · Feb 20, 2022 · 7223fc8 · 7223fc8
2 parents 17490b1 + 36fa84d
commit 7223fc8
Show file tree

Hide file tree

Showing 10 changed files with 3 additions and 3 deletions.
diff --git a/examples/benchmark/BUILD → tools/cpp_benchmark/BUILD b/examples/benchmark/BUILD → tools/cpp_benchmark/BUILD
@@ -1,7 +1,7 @@
 package(default_visibility = ["//visibility:public"])
 
 cc_binary(
-    name = "benchmark",
+    name = "cpp_benchmark",
     srcs = [
         "main.cpp",
         "timer.h",

diff --git a/examples/benchmark/README.md → tools/cpp_benchmark/README.md b/examples/benchmark/README.md → tools/cpp_benchmark/README.md
@@ -9,13 +9,13 @@ Run with bazel:
 > Note: Make sure libtorch and TensorRT are in your LD_LIBRARY_PATH before running, if you need a location you can `export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:[WORKSPACE ROOT]/bazel-Torch-TensorRT/external/libtorch/lib:[WORKSPACE ROOT]/bazel-Torch-TensorRT/external/tensorrt/lib`
 
 ``` sh
-bazel run //cpp/benchmark --cxxopt="-DNDEBUG" --cxxopt="-DJIT" --cxxopt="-DTRT" -- [PATH TO JIT MODULE FILE] [INPUT SIZE (explicit batch)]
+bazel run //tools/cpp_benchmark --cxxopt="-DNDEBUG" --cxxopt="-DJIT" --cxxopt="-DTRT" -- [PATH TO JIT MODULE FILE] [INPUT SIZE (explicit batch)]
 ```
 
 For example:
 
 ``` shell
-bazel run //cpp/benchmark  --cxxopt="-DNDEBUG" --cxxopt="-DJIT" --cxxopt="-DTRT" -- $(realpath /tests/models/resnet50.jit.pt) "(32 3 224 224)"
+bazel run //tools/cpp_benchmark  --cxxopt="-DNDEBUG" --cxxopt="-DJIT" --cxxopt="-DTRT" -- $(realpath /tests/models/resnet50.jit.pt) "(32 3 224 224)"
 ```
 
 ### Options

diff --git a/examples/benchmark/main.cpp → tools/cpp_benchmark/main.cpp b/examples/benchmark/main.cpp → tools/cpp_benchmark/main.cpp
diff --git a/examples/benchmark/timer.h → tools/cpp_benchmark/timer.h b/examples/benchmark/timer.h → tools/cpp_benchmark/timer.h
diff --git a/examples/benchmark/py/README.md → tools/perf/README.md b/examples/benchmark/py/README.md → tools/perf/README.md
diff --git a/examples/benchmark/py/config/vgg16.yml → tools/perf/config/vgg16.yml b/examples/benchmark/py/config/vgg16.yml → tools/perf/config/vgg16.yml
diff --git a/examples/benchmark/py/config/vgg16_trt.yml → tools/perf/config/vgg16_trt.yml b/examples/benchmark/py/config/vgg16_trt.yml → tools/perf/config/vgg16_trt.yml
diff --git a/examples/benchmark/py/models/.gitkeep → tools/perf/models/.gitkeep b/examples/benchmark/py/models/.gitkeep → tools/perf/models/.gitkeep
diff --git a/examples/benchmark/py/perf_run.py → tools/perf/perf_run.py b/examples/benchmark/py/perf_run.py → tools/perf/perf_run.py
diff --git a/examples/benchmark/py/requirements.txt → tools/perf/requirements.txt b/examples/benchmark/py/requirements.txt → tools/perf/requirements.txt