pytorch/benchmark

Pull Requests Commits

[optim][ub] Skip CPU benchmarks since we don't observe their stats anyways

janeyx99 committed 2 years ago

114ec8b1

Print torch git version (commit) in workflow (#1779)

janeyx99 committed 2 years ago

d9e6653a

fix uninitialized model_flop variable issue (#1778)

snadampal committed 2 years ago

c03597e8

llama v2 (#1775)

msaroufim committed 2 years ago

196b3b8a

Add functionality to TRT backend (#1753)

gs-olive committed 2 years ago

30a3c4a9

Get PT-D benchmark to support being run internally

H-Huang committed 2 years ago

36e1ed1c

Add torch.utils.flop_counter to TorchBench

monicatang committed 2 years ago

1dada69d

Fix fvcore flops counting for torchvision models

xuzhao9 committed 2 years ago

9d84f9e6

Add dependencies in the base docker build (#1776)

xuzhao9 committed 2 years ago

79bc754f

Fix require_grad typo (#1771)

kit1980 committed 2 years ago

ae8f06fe

Remove unnecessary model attribute assignment on 'freqs_cis' (#1766)

BowenBao committed 2 years ago

e8c1cf07

Add HF Auth mixin to Stable Diffusion (#1763)

msaroufim committed 2 years ago

411e3881

add API coverage test (#1394)

FindHao committed 2 years ago

09de70c8

FIX SAM for bfloat16 (#1764)

msaroufim committed 2 years ago

745644f3

remove wrong reference to __init__.py (#1746)

janeyx99 committed 2 years ago

2ea018e9

export chrome trace when profiler is enabled (#1760)

dshi7 committed 2 years ago

bfb6d156

fix broken torchbench mimo_cmf_30x

dshi7 committed 2 years ago

48a8ef6d

Move lit_llama and variants to canary_models (#1761)

ezyang committed 2 years ago

16e235ea

Add test runner on ibm cloud (#1755)

spzala committed 2 years ago

3df1cc93

Add lit-llama benchmarks (logits, autoregressive generation, lora fine tuning) (#1730)

ezyang committed 2 years ago

02ff72b4

Add special device mapping (#1752)

FindHao committed 2 years ago

9df52151

Increase the worker timeout to 1800 seconds

xuzhao9 committed 2 years ago

ae3534ab

Fix broken unit test (#1743)

xuzhao9 committed 2 years ago

4e9d86ec

Add boto3 to requirements.txt (#1745)

Thiago Crepaldi committed 2 years ago

5b2a70ea

Refactor Execution Graph (EG) to Execution Trace (ET) (#1566)

louisfeng committed 2 years ago

6d84522e

Add dependency boto3 to readme.md (#1741)

FindHao committed 2 years ago

eadfa982

Add back video_watch

xuzhao9 committed 2 years ago

b90583ac

Add GCP A100 bisection workflow (#1742)

xuzhao9 committed 2 years ago

81508866

Minor fix of the nvidia driver install (#1740)

xuzhao9 committed 2 years ago

84bac2a9

Upgrade the NVidia kernel driver (#1739)

xuzhao9 committed 2 years ago

2b9de6b0

Older