saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/benchmarks
History
Vasiliy Kuznetsov 2773ed3082 hardswish: remove unnecessary quantize call (#36980) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36980

Missed this on the original diff, fixing.  Create the output tensor directly instead of quantizing it.

Test Plan:
tests still pass
microbenchmarks show a 2x performance improvment for int8:
https://gist.github.com/vkuzo/3b321b428e4c38e805000961c263286b (this
will depend on input size)

Imported from OSS

Differential Revision: D21185970

fbshipit-source-id: 5b9e93d9f9ac05a8120532bd03ad347541a132c2
2020-04-22 16:15:54 -07:00
..
distributed/ddp Add distributed data parallel benchmark tool (#35198) 2020-04-08 15:07:03 -07:00
fastrnns Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
framework_overhead_benchmark Fix spelling errors 2020-01-28 04:46:15 -08:00
operator_benchmark hardswish: remove unnecessary quantize call (#36980) 2020-04-22 16:15:54 -07:00
overrides_benchmark [RELAND] Add __torch_function__ benchmarks (#36138) 2020-04-10 09:14:31 -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

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