frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
21a9a93eb4bf8660e24721bb24f69ebb0340f2fb
pytorch/benchmarks
History
Hui Guo 2a53897114 [jit][tensorexpr] Added aten::batch_norm into fuser when in inference mode (#54204) 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54204

Test Plan: Imported from OSS

Reviewed By: bertmaher

Differential Revision: D27134348

Pulled By: huiguoo

fbshipit-source-id: 5ea7a6c5bc694fcdfc436dba3fa6eb269420324e
2021-03-23 04:41:52 -07:00
..
cpp/tensorexpr
[jit][tensorexpr] Added aten::batch_norm into fuser when in inference mode (#54204)
2021-03-23 04:41:52 -07:00
distributed
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
fastrnns
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
framework_overhead_benchmark
Remove py2 compatible future imports (#44735)
2020-09-16 12:55:57 -07:00
functional_autograd_benchmark
Fix typo in torchvision_models.py (#53968)
2021-03-15 11:02:06 -07:00
operator_benchmark
fix broken quantization_test in operator_benchmark (#53153)
2021-03-08 12:12:57 -08:00
overrides_benchmark
Add __torch_function__ for methods (#37091)
2020-08-05 20:44:13 -07:00
profiler_benchmark
Use libkineto in profiler (#46470)
2020-11-25 04:32:16 -08:00
record_function_benchmark
Fix D23995953 import.
2020-09-29 19:30:23 -07:00
serialization
[JIT] Make new zip serialization for torch save/load significantly (~70%) faster (#38379)
2020-05-29 01:56:18 -07:00
sparse/dlmc
matmul performance benchmarks (#51647)
2021-03-14 00:25:45 -08:00
static_runtime
[Static Runtime] Fix bug in reshape_copy (#54467)
2021-03-22 22:20:55 -07:00
tensorexpr
[NNC] Implementation for aten::cat without conditionals. (#53128)
2021-03-07 22:57:02 -08:00
compare-fastrnn-results.py
Benchmarks: add scripts for FastRNNs results comparison. (#44134)
2020-09-03 13:44:42 -07:00
compare.sh
Benchmarks: add scripts for FastRNNs results comparison. (#44134)
2020-09-03 13:44:42 -07:00
README.md
Fix spelling errors
2020-01-28 04:46:15 -08:00
upload_scribe.py
Benchmarks: make fuser and executor configurable from command line. (#44291)
2020-09-09 11:59:35 -07:00

README.md

PyTorch Benchmarks

NOTE: This folder is currently work in progress.

This folder contains scripts that produce reproducible timings of various PyTorch features.

It also provides mechanisms to compare PyTorch with other frameworks.

Setup environment

Make sure you're on a machine with CUDA, torchvision, and pytorch installed. Install in the following order:

# Install torchvision. It comes with the pytorch stable release binary
conda install pytorch torchvision -c pytorch

# Install the latest pytorch master from source.
# It should supersede the installation from the release binary.
cd $PYTORCH_HOME
python setup.py build develop

# Check the pytorch installation version
python -c "import torch; print(torch.__version__)"

Benchmark List

Please refer to each subfolder to discover each benchmark suite