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
fe7e1bd1ceee405eb14ba54993ea90ddb463cf08
pytorch/benchmarks
History
Don Jang fe7e1bd1ce [Static Runtime] Add auto-generated out variant dispatchers (#72603) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72603

This change adds out variant dispatchers generated by the previous diff.

The number of the out variant dispatchers generated by this diff is 133, which increases the out variant coverage by 309% (current: 43, this diff: 133 + 43 = 176). This number is expected to increase a lot as we develop this script further to cover more ops.

Test Plan:
**Unittest**
Confirmed
```
buck run //caffe2/benchmarks/static_runtime:static_runtime_cpptest
```
is passing.

Reviewed By: swolchok

Differential Revision: D33373928

fbshipit-source-id: 4d94d788282f3f313bb36f2f9452edecd9862246
(cherry picked from commit e4ce8b386d1fcc47b86cb9c9016a70e7a31b452c)
2022-02-28 08:39:10 +00:00
..
cpp
Nvfuser code bump 2_1_2022 (#72127)
2022-02-15 00:43:16 +00:00
distributed
Remove .data from benchmarks and tensorboard (#65389)
2021-09-22 11:16:59 -07:00
fastrnns
Remove .data from benchmarks and tensorboard (#65389)
2021-09-22 11:16:59 -07:00
framework_overhead_benchmark
Remove py2 compatible future imports (#44735)
2020-09-16 12:55:57 -07:00
functional_autograd_benchmark
Extend autograd functional benchmarking to run vectorized tasks (#67045)
2021-12-21 17:20:29 -08:00
fuser
Benchmarks for various fusers (#67622)
2021-11-04 18:57:17 -07:00
instruction_counts
Allow instruction counting to use shared memory as a staging ground. (And a couple other tweaks.) (#56711)
2021-05-12 20:37:41 -07:00
operator_benchmark
Fix command example (#72847)
2022-02-16 21:45:45 +00:00
overrides_benchmark
Use classmethods for overrides (#64841)
2021-09-17 08:32:49 -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
Add CSR (compressed sparse row) layout for sparse tensors (#50937)
2021-04-12 10:09:12 -07:00
static_runtime
[Static Runtime] Add auto-generated out variant dispatchers (#72603)
2022-02-28 08:39:10 +00:00
tensorexpr
[nnc] Added micro-benchmark to show perf improvement with cat subgraph optimization (#59581)
2021-06-18 14:32:09 -07: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
Add CSR (compressed sparse row) layout for sparse tensors (#50937)
2021-04-12 10:09:12 -07:00
upload_scribe.py
Fix benchmark's import module and remove its usage of tools.stats.scribe (#61808)
2021-07-19 09:45:05 -07:00

README.md

PyTorch Benchmarks

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