pytorch/benchmark

Pull Requests Commits

[optim][ub] Add stable diffusion models to dense to skip on sparseadam

janeyx99 committed 2 years ago

cc4a35ea

Update README.md (#1920)

msaroufim committed 2 years ago

2bd1aa13

Add interface to support multiple batches (#1914)

xuzhao9 committed 2 years ago

842f11db

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

Add a simple tool to generate torchbench model test cases with input model spec

xuzhao9 committed 2 years ago

535cfff0

Add pt2 graph breaks (#1858)

xuzhao9 committed 2 years ago

df4da9bd

Older