pytorch/benchmark

Pull Requests Commits

Add readme with usage & example parse_ddp.csv output

davidberard98 committed 3 years ago

022ef992

skip cuda on hf_GPT2_large to avoid OOM

davidberard98 committed 3 years ago

55e81407

remove userbenchmark/distributed/ddp_experiments.py

davidberard98 committed 3 years ago

d32712a2

move userbenchmark/distributed -> userbenchmark/ddp_experiments

davidberard98 committed 3 years ago

405a9007

wconstab committed 3 years ago

e241c722

skip timm_vit_large on cuda CI since it ooms

wconstab committed 3 years ago

5140f9f4

wconstab committed 3 years ago

e3daa21e

Add FSDP support (untested)

wconstab committed 3 years ago

f18ebcfa

Added Large Models (GPT, T5, TIMM-VIT)

sanketpurandare committed 3 years ago

2cfad266

Confiure TB for ddp experiments

wconstab committed 3 years ago

79bf4084

Add train test for cmf_10x, ads_dhen_5x, umia_v1_fbr and blue_reels_vdd_v2 (#1209)

xuzhao9 committed 3 years ago

52c41137

Improve bisection workflow (#1210)

xuzhao9 committed 3 years ago

8c71f407

Wording and nit updates to README and ADDING_MODELS documentation (#1202)

adnanaziz committed 3 years ago

1965927b

Update TensorRT version (#1206)

xuzhao9 committed 3 years ago

29328ea6

Remove lazy tensor workflow and move benchmark script to userbenchmark (#1200)

xuzhao9 committed 3 years ago

9d66f4f0

maml_omniglot model with functorch (#1179)

zou3519 committed 3 years ago

c80d22b7

Add lennard-jones model (#1180)

zou3519 committed 3 years ago

e64fc735

Add pure-functorch variant of the dp-cifar10 model (#1178)

zou3519 committed 3 years ago

bb3a8be2

Fix readme --distributed -> --model (#1187)

erichan1 committed 3 years ago

8045bb7d

Fix Huggingface model issue with distributed (#1189)

xuzhao9 committed 3 years ago

26e450c6

Fix CI errors and add reload when detecting pytorch version inconsistency (#1176)

xuzhao9 committed 3 years ago

f9189cb9

Create README for distributed benchmark (#1183)

erichan1 committed 3 years ago

67c6d71c

Fix deepspeed for bert and t5 (#1181)

erichan1 committed 3 years ago

5ad56723

Add fsdp to e2e hf_bert (#1177)

erichan1 committed 3 years ago

863959ec

Add DALL-E 2 model to pytorch/benchmark (#1151)

nmacchioni committed 3 years ago

0457a650

Add FSDP to T5 (#1142)

erichan1 committed 3 years ago

73ebe8ab

Removing .github/scale-config.yml, now this repo is using the config in test-infra (#1172)

jeanschmidt committed 3 years ago

d2a35a5f

Remove the use of niter in the model code (#1171)

xuzhao9 committed 3 years ago

72bee5cd

Move workflows to CUDA 11.6 (#1162)

xuzhao9 committed 3 years ago

3b91b5a6

Fix a device inconsistency in dlrm (#1170)

desertfire committed 3 years ago

24b95f2f

Older