Google Git
Sign in
android / platform / external / pytorch / 6386e4d244 / . / benchmarks / operator_benchmark
tree: 7d0a07f007485dfb8134d99c3e927fc8e884237e [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