pytorch/benchmark

Pull Requests Commits

add funcs option, realized I should really subscribe to torchbench stuff

janeyx99 committed 2 years ago

13a2e396

draft, in progress

janeyx99 committed 2 years ago

bf9e8057

we get a json now

janeyx99 committed 2 years ago

8da0e60e

make script more advanced

janeyx99 committed 2 years ago

5e6af483

janeyx99 committed 2 years ago

07adacfc

janeyx99 committed 2 years ago

26f93d87

[WIP] prelim script to run some models

janeyx99 committed 2 years ago

47734dfe

Fix the case when workflow is empty (#1438)

xuzhao9 committed 2 years ago

c9e3dd95

Add the --mps_only option. (#1432)

kulinseth committed 2 years ago

1635a749

Disable the nvfuser nightly CI (#1431)

xuzhao9 committed 2 years ago

4fb6838c

Fix torchrec_dlrm model and enable fbcode test (#1428)

xuzhao9 committed 2 years ago

08fc5914

Use the correct batch size for timm ViT to improve stableness (#1419)

xuzhao9 committed 2 years ago

06538fd1

Upgrade to gcc-11 in the A100 Docker to fix CI error (#1427)

xuzhao9 committed 2 years ago

e2dfe592

Add torchtext to fix benchmark installation error (#1424)

spzala committed 2 years ago

0e6a7de9

Upgrade the actions used by the bisection workflow (#1417)

xuzhao9 committed 2 years ago

d2c692c2

replace torchbench impelmentation with torchrec that shards the module (#1401)

colin2328 committed 2 years ago

9806a7ec

Fix vision model on inductor and profile on CPU only machines (#1415)

xuzhao9 committed 2 years ago

7bd50b01

change fastNLP_Bert's task name (#1410)

FindHao committed 2 years ago

3fe6446f

Add detailed task names (#1411)

Yueming Hao committed 2 years ago

74895429

Add channels_last for all torchbench models and enable it with run.py (#1371)

chuanqi129 committed 2 years ago

d64579cc

Pin pygame version to fix soft_actor_critic (#1412)

xuzhao9 committed 2 years ago

43ee38aa

Match AITModule on ait inputs and fx inputs (#262)

tissue3 committed 2 years ago

1bbb66c4

Fix type in iter rounds (#1408)

xuzhao9 committed 2 years ago

411117d8

Add mimo_cmf_30x

bertmaher committed 2 years ago

bf02c40a

Upgrade to upload-artifact v3 to resolve GitHub warnings (#1407)

xuzhao9 committed 2 years ago

91d91104

Force creation of the conda environment and upgrade dependencies (#1406)

xuzhao9 committed 2 years ago

adce3da2

Add CI to main branch commits (#1404)

xuzhao9 committed 2 years ago

916ab70b

Fix A100 workflow (#1403)

xuzhao9 committed 2 years ago

4a8f9927

Upgrade to CUDA 11.7 and remove V1 workflows (#1398)

xuzhao9 committed 2 years ago

11f7141d

Fix a bug in train correctness checking

xuzhao9 committed 2 years ago

0c99340f

Older