pytorch/benchmarks/operator_benchmark
Mingzhe Li 26f12af537 Fix op benchmarks error in OSS environment (#19518)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/19518

Previous design needs to run the op benchmarks from PyTorch root directory which could lead to `module not found` error in OSS environment. This diff fixes that issue by making the benchmark to be launched in the `benchmarks` folder.

Reviewed By: ilia-cher

Differential Revision: D15020787

fbshipit-source-id: eb09814a33432a66cc857702bc86538cd17bea3b
2019-04-19 16:25:16 -07:00
..
ops Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
__init__.py Operator-level performance microbenchmarks (#18740) 2019-04-02 17:06:19 -07:00
batchnorm_benchmark.py Add a fast path for batch-norm CPU inference. (#19152) 2019-04-16 19:27:54 -07:00
benchmark_caffe2.py Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
benchmark_core.py Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
benchmark_pytorch.py Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
benchmark_runner.py Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
benchmark_test_generator.py Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00
benchmark_utils.py make separate operators as independent binaries (#19450) 2019-04-18 20:00:47 -07:00
README.md Fix op benchmarks error in OSS environment (#19518) 2019-04-19 16:25:16 -07:00

PyTorch/Caffe2 Operator Micro-benchmarks

Run benchmarks

Go to pytorch-src-folder/benchmarks python -m operator_benchmark.ops.matmul_test

should report the execution time of matmul operator with PyTorch and Caffe2