Google Git
Sign in
android / platform / external / pytorch / 63a7c7bb2a / . / benchmarks / operator_benchmark
tree: 38b3fe1dae2d9828ab60603e5a61254771f15c63 [path history] [tgz]
  1. c2/
  2. common/
  3. pt/
  4. __init__.py
  5. benchmark_all_test.py
  6. benchmark_caffe2.py
  7. benchmark_core.py
  8. benchmark_pytorch.py
  9. benchmark_runner.py
  10. benchmark_test_generator.py
  11. benchmark_utils.py
  12. operator_benchmark.py
  13. README.md
benchmarks/operator_benchmark/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