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