diff --git a/examples/benchmark/BUILD b/tools/cpp_benchmark/BUILD similarity index 89% rename from examples/benchmark/BUILD rename to tools/cpp_benchmark/BUILD index 7ee021ea21..8aa36b0e90 100644 --- a/examples/benchmark/BUILD +++ b/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 b/tools/cpp_benchmark/README.md similarity index 78% rename from examples/benchmark/README.md rename to tools/cpp_benchmark/README.md index 3a264c9544..efbd1d5e14 100644 --- a/examples/benchmark/README.md +++ b/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 b/tools/cpp_benchmark/main.cpp similarity index 100% rename from examples/benchmark/main.cpp rename to tools/cpp_benchmark/main.cpp diff --git a/examples/benchmark/timer.h b/tools/cpp_benchmark/timer.h similarity index 100% rename from examples/benchmark/timer.h rename to tools/cpp_benchmark/timer.h diff --git a/examples/benchmark/py/README.md b/tools/perf/README.md similarity index 100% rename from examples/benchmark/py/README.md rename to tools/perf/README.md diff --git a/examples/benchmark/py/config/vgg16.yml b/tools/perf/config/vgg16.yml similarity index 100% rename from examples/benchmark/py/config/vgg16.yml rename to tools/perf/config/vgg16.yml diff --git a/examples/benchmark/py/config/vgg16_trt.yml b/tools/perf/config/vgg16_trt.yml similarity index 100% rename from examples/benchmark/py/config/vgg16_trt.yml rename to tools/perf/config/vgg16_trt.yml diff --git a/examples/benchmark/py/models/.gitkeep b/tools/perf/models/.gitkeep similarity index 100% rename from examples/benchmark/py/models/.gitkeep rename to tools/perf/models/.gitkeep diff --git a/examples/benchmark/py/perf_run.py b/tools/perf/perf_run.py similarity index 100% rename from examples/benchmark/py/perf_run.py rename to tools/perf/perf_run.py diff --git a/examples/benchmark/py/requirements.txt b/tools/perf/requirements.txt similarity index 100% rename from examples/benchmark/py/requirements.txt rename to tools/perf/requirements.txt