mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
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 |
||
|---|---|---|
| .. | ||
| ops | ||
| __init__.py | ||
| batchnorm_benchmark.py | ||
| benchmark_caffe2.py | ||
| benchmark_core.py | ||
| benchmark_pytorch.py | ||
| benchmark_runner.py | ||
| benchmark_test_generator.py | ||
| benchmark_utils.py | ||
| README.md | ||
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