pytorch/benchmarks
Vasiliy Kuznetsov cc78914755 qactivation_benchmarks: small bug fix (#35731)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35731

Changes relu and relu6 to point to the functional implementations here.
The previous behavior tested the time to create the module, but didn't actually run the
function (I noticed this when adding the new input sizes and seeing
the measured time not change).

Test Plan:
run the benchmark, the time now changes as expected with input size for
these.

Imported from OSS

Differential Revision: D20875542

fbshipit-source-id: 3a6278a7a861437d613c1e30698a58175a8e8555
2020-04-06 15:02:33 -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 qactivation_benchmarks: small bug fix (#35731) 2020-04-06 15:02:33 -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 [TensorExpr] Fix imports in tensorexpr benchmarks. (#35830) 2020-04-01 14:23:33 -07:00
README.md Fix spelling errors 2020-01-28 04:46:15 -08:00

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