saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/benchmarks
History
Michael Suo 6491bf2855 Revert D20777341: [pytorch][PR] Add __torch_function__ benchmarks. 
Test Plan: revert-hammer

Differential Revision:
D20777341

Original commit changeset: 6aaaf2a07553

fbshipit-source-id: 1c324f91f85ac624bf878297c96c682a46958954
2020-04-01 10:23:00 -07:00
..
fastrnns Add tensorexpr benchmarks. (#35064) 2020-03-20 12:01:31 -07:00
framework_overhead_benchmark Fix spelling errors 2020-01-28 04:46:15 -08:00
operator_benchmark add quantized version of hardswish operator (#34820) 2020-03-24 15:16:58 -07:00
profiler_benchmark [profiler][rpc] fix a race condition in the profiler when multiple threads call (#33719) 2020-03-16 18:41:16 -07:00
tensorexpr Expose enablement of TensorExpr fuser as env variable (#35341) 2020-03-26 14:31:57 -07:00
README.md Fix spelling errors 2020-01-28 04:46:15 -08: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