pytorch/benchmark

Pull Requests Commits

Update model_factory.py

msaroufim committed 2 years ago

Verified 1e7f23f6

Update model_factory.py

msaroufim committed 2 years ago

Verified f2ff5368

Update __init__.py

msaroufim committed 2 years ago

Verified 12df4dfc

msaroufim committed 2 years ago

beceab38

Support llamav27b training

msaroufim committed 2 years ago

b7ed2305

Add nanogpt_train to torchbench, and merge with nanogpt_generate into joint folder (#1911)

xmfan committed 2 years ago

d00b9421

Explain `test.py`'s `NotImplementedError`s (#1913)

hmellor committed 2 years ago

1f70eae8

Add warmup and benchamrk iterations for cpu userbenchmark (#1908)

chuanqi129 committed 2 years ago

2d05303a

Rename the `enable_fp16_half()` function to `enable_fp16` (#1900)

xuzhao9 committed 2 years ago

644a9ceb

Pin beartype for torchbench (#1907)

malfet committed 2 years ago

8360dba3

Fix ig_ctr model failure on AIBench

xuzhao9 committed 2 years ago

cccaacee

Stable Diffusion XL (#1882)

msaroufim committed 2 years ago

95207dfd

Skip the model test by the device name in its metadata. (#1906)

xuzhao9 committed 2 years ago

568f9475

Upgrade version of transformers (#1905)

xuzhao9 committed 2 years ago

8d159fda

`norm()` -> `linalg.norm()` in `lennard_jones` (#1898)

hmellor committed 2 years ago

22cb901a

Remove `.cuda()` from `cm3leon`'s `model.py` (#1897)

hmellor committed 2 years ago

6532c61f

Disable Deepcopy on stable_diffusion_unet. (#1895)

xuzhao9 committed 2 years ago

e1b51f52

Always load the model within the same process in sandcastle test.

xuzhao9 committed 2 years ago

0f33c9b7

Exclude timm_vision_transformer pt2, fix runtime errors in 1888 (#1890)

janeyx99 committed 2 years ago

fd78a760

SD: image encoder bench (#1881)

msaroufim committed 2 years ago

9cbeee3b

Disable torchtext and related models in v2 workflow (#1877)

xuzhao9 committed 2 years ago

feb15ff9

the accuracy test always returns none for the output

Lu Fang committed 2 years ago

11183324

Add DTensor LLaMA inference model: simple_gpt (#1867)

xmfan committed 2 years ago

0a64c552

Fix doctr model build failure (#1883)

xuzhao9 committed 2 years ago

e768fd35

Distinguish compile time from run time (#1865)

janeyx99 committed 2 years ago

1e7bcd0e

Fix the test name in readme (#1878)

xuzhao9 committed 2 years ago

ffbbebb9

Move AOTInductor runtime headers

desertfire committed 2 years ago

ade7d12d

Remove conda forge dependencies. (#1860)

xuzhao9 committed 2 years ago

1a17fc9d

Del `(object)` from 10 inc pytorch/benchmark/torchbenchmark/models/LearningToPaint/baseline_modelfree/DRL/evaluator.py

r-barnes committed 2 years ago

31444e4e

Del `(object)` from 10 inc pytorch/benchmark/torchbenchmark/models/moco/main_lincls.py

r-barnes committed 2 years ago

273471c7

Older