pytorch/benchmark

Pull Requests Commits

testing changes

davidberard98 committed 3 years ago

3b183535

impl using FileStore

davidberard98 committed 3 years ago

fbee639e

partial impl of reusing allocation & attempt at implementing my own barrier

davidberard98 committed 3 years ago

acdb6179

unlandable changes

davidberard98 committed 3 years ago

5f8496a8

ddp+dynamo: assorted fixes

davidberard98 committed 3 years ago

c1edf77d

Implemented staged tests for train

xuzhao9 committed 3 years ago

9ca9f58e

Enable CI on AWS T4 Metal instance (#1236)

xuzhao9 committed 3 years ago

d6a3e064

functorch userbenchmark (#1235)

zou3519 committed 3 years ago

147390dc

disable torchinductor.config.triton.use_bmm with tritonmm=triton option

jspark1105 committed 3 years ago

5814dbcc

Migrate instruction count benchmark to userbenchmark (#1195)

xuzhao9 committed 3 years ago

1ea5f44c

Fix broken ci from gpt2_large (#1230)

xuzhao9 committed 3 years ago

20686776

Add flag for dynamo+ddp optimizations (#1221)

davidberard98 committed 3 years ago

c8133956

Pin pycocotools version to make the patch work (#1224)

xuzhao9 committed 3 years ago

d0bf8a65

change training precision default to tf32

jspark1105 committed 3 years ago

628ee78c

Another fix for the bisection job (#1227)

xuzhao9 committed 3 years ago

45f0678e

Update ddp+dynamo benchmark arguments (#1222)

davidberard98 committed 3 years ago

48ca7613

Load instance setup script before running benchmark (#1217)

xuzhao9 committed 3 years ago

b96d07a9

Add torchinductor triton config mm

Mengchi Zhang committed 3 years ago

c3813be9

add infra and models for ddp experiments (#1190)

wconstab committed 3 years ago

37e714b0

Support customized code evaluation in torchdynamo options

xuzhao9 committed 3 years ago

b18d4f6f

Allow Ubuntu in the machine type. (#1214)

xuzhao9 committed 3 years ago

bb0ef3d3

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

Older