Google Git
Sign in
android / platform / external / pytorch / f894ef7263 / . / benchmarks / operator_benchmark
tree: 1495d4e688b51e3c4172bfec9b51ca0ca17b1014 [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