pytorch/benchmark

Pull Requests Commits

Add get/set_optimizer to E2EBenchmarkModel API

janeyx99 committed 2 years ago

9caa443e

Add instructions to test model after submitting (#1447)

msaroufim committed 2 years ago

e080877d

GeForce clock query fix Issue #1440 (#1445)

Mike Schneider committed 2 years ago

9ab7a6f2

enable peak memory measurement by default (#1392)

FindHao committed 2 years ago

1604885b

Add the MPS device to eval mode. (#1435)

kulinseth committed 2 years ago

e9941f39

Clarify instllation step (#1426)

spzala committed 2 years ago

f33ce935

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

Print grad values when not matching for debugging purpose (#1399)

brad-mengchi committed 2 years ago

1793f37f

Older