pytorch/benchmark

Pull Requests Commits

fix lazy_bench.py precision option

Krovatkin committed 3 years ago

edeb0870

Add FAMBench dlrm model (#878)

xuzhao9 committed 3 years ago

9d8e569c

Add pplbench beanmachine model (#869)

agunapal committed 3 years ago

97efd8b0

Fix PyTorch unet train, only use mixed precision in forward pass. (#873)

xuzhao9 committed 3 years ago

6b13e204

Add more NVFuser microbenchmarks (#801)

davidberard98 committed 3 years ago

366b1843

Migrate from Ubuntu 16.04 to 20.04 (#765)

kit1980 committed 3 years ago

07a4914c

move torchrec dlrm example into facebookresearch/dlrm (#221)

colin2328 committed 3 years ago

c5b02e7d

Delete dead Py2 code from pytorch/benchmark (#866)

lisroach committed 3 years ago

acac953b

Add CI test on compiler backends (#832)

xuzhao9 committed 3 years ago

e990e696

Fix demucs jit error with type annotation (#856)

xuzhao9 committed 3 years ago

fbd17497

adding lazy_bench tests (#846)

Krovatkin committed 3 years ago

7c6da287

nvfuser microbenches - import utils from torch.utils (#854)

davidberard98 committed 3 years ago

c4710504

Porting fixes from wconstab/ltc to main (#851)

Krovatkin committed 3 years ago

c60f7edd

Add FAMBench xlmr model (#835)

xuzhao9 committed 3 years ago

032f079f

Improve the `run_sweep.py` script to support precision attribute (#844)

xuzhao9 committed 3 years ago

356fd5da

Add accelerate to requirements.txt (#849)

mrshenli committed 3 years ago

86d29d08

Upgrade to cudnn 8.3.2 (#847)

xuzhao9 committed 3 years ago

ff711465

Pin cmake version because recent cmake upgrade breaks pytorch (#845)

xuzhao9 committed 3 years ago

f1c00d01

Fix 'RecursiveScriptModule' object has no attribute 'forward' in demucs (#564)

jansel committed 3 years ago

ce5186d3

Fix `get_module()` bug in timm_efficientdet (#842)

xuzhao9 committed 3 years ago

b1e7fe8b

Remove print from speech_transformer (#837)

jansel committed 3 years ago

0c840c14

Disable moco on CPU (#838)

jansel committed 3 years ago

64268fac

Cache the huggingface model at install stage (#836)

xuzhao9 committed 3 years ago

332055b4

Add initial fambench support to e2e_models (#821)

xuzhao9 committed 3 years ago

4809ecf6

Refactor the huggingface model code (#825)

xuzhao9 committed 3 years ago

cf34503e

Adding the model input iterators. (#806)

xuzhao9 committed 3 years ago

c0c09d4d

Add the efficientdet model (#582)

xuzhao9 committed 3 years ago

38e31bdc

Fix GPU memory leak introduced by correctness checking (#830)

xuzhao9 committed 3 years ago

03372976

fix lower_to_trt api (#822)

Wei Wei committed 3 years ago

21cfa45f

Don't load empty directory in unit tests (#823)

xuzhao9 committed 3 years ago

f8ab5657

Older