pytorch/benchmark

Pull Requests Commits

don't wrap results in dict

davidberard98 committed 3 years ago

411c3a8d

correctly label # of nodes when showing jobid

davidberard98 committed 3 years ago

da079a7c

davidberard98 committed 3 years ago

b1ac873b

make FileBarrier accept timeout as argument

davidberard98 committed 3 years ago

e80dd0fa

Run multiple jobs per allocation

davidberard98 committed 3 years ago

ff333ab4

Reduce profiling runs on distributed models to 2

davidberard98 committed 3 years ago

9d002a1d

unlandable changes

davidberard98 committed 3 years ago

0b03ab66

correctness check train

jspark1105 committed 3 years ago

65e4f449

Upgrade to checkout@v3 (#1240)

xuzhao9 committed 3 years ago

e403c97e

Required changes after cuBLAS workspace allocation change (#1226)

eqy committed 3 years ago

c5879304

Fix name of the platform (#1241)

xuzhao9 committed 3 years ago

5a89750b

Add PYTORCH_TEST_SKIP_CUDA_MEM_LEAK_CHECK (#1237)

sangongs committed 3 years ago

f5f795bb

ddp+dynamo: assorted fixes (#1223)

davidberard98 committed 3 years ago

f320cda1

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

Older