pytorch/benchmark

Pull Requests Commits

xuzhao9 committed 3 years ago

9ec7b658

Dont' import amp when not needed

xuzhao9 committed 3 years ago

8f82c6e2

xuzhao9 committed 3 years ago

ae105989

Add copy of rnnt train

xuzhao9 committed 3 years ago

70efdeb1

Only enable optimizer when amp is enabled and apex is available.

xuzhao9 committed 3 years ago

a347567b

Fix other train issues

xuzhao9 committed 3 years ago

6bec36b9

More refinements.

xuzhao9 committed 3 years ago

bb583816

Finished train loop

xuzhao9 committed 3 years ago

e9a9d9bb

Add train config

xuzhao9 committed 3 years ago

bdf80e70

Finished eval, working on train

xuzhao9 committed 3 years ago

97499dab

Working on inference

xuzhao9 committed 3 years ago

1fda9cea

Initial rnnt commit

xuzhao9 committed 3 years ago

9e1652de

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 4 years ago

07a4914c

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

colin2328 committed 4 years ago

c5b02e7d

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

lisroach committed 4 years ago

acac953b

Add CI test on compiler backends (#832)

xuzhao9 committed 4 years ago

e990e696

Fix demucs jit error with type annotation (#856)

xuzhao9 committed 4 years ago

fbd17497

adding lazy_bench tests (#846)

Krovatkin committed 4 years ago

7c6da287

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

davidberard98 committed 4 years ago

c4710504

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

Krovatkin committed 4 years ago

c60f7edd

Add FAMBench xlmr model (#835)

xuzhao9 committed 4 years ago

032f079f

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

xuzhao9 committed 4 years ago

356fd5da

Add accelerate to requirements.txt (#849)

mrshenli committed 4 years ago

86d29d08

Upgrade to cudnn 8.3.2 (#847)

xuzhao9 committed 4 years ago

ff711465

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

xuzhao9 committed 4 years ago

f1c00d01

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

jansel committed 4 years ago

ce5186d3

Older