pytorch/benchmark

Pull Requests Commits

Make it easier to add more backends (#1059)

xuzhao9 committed 3 years ago

d9fa6b8d

Add `--output` option to userbenchmark/nvfuser (#1062)

xwang233 committed 3 years ago

305152cf

Upload userbenchmark result to Scribe (#1061)

xuzhao9 committed 3 years ago

654a7f6c

Remove niter for models (#1054)

xuzhao9 committed 3 years ago

b743a9bb

fix a printout bug; remove default memory monitoring (#1049)

FindHao committed 3 years ago

7bdb8322

Enable 'binned' dynamic shapes for hf models (#1050)

wconstab committed 3 years ago

2ebc1c03

Support probing the best batch size for inference tests (#1027)

FindHao committed 3 years ago

4a35c22a

fix batch size problem (#1045)

zzpmiracle committed 3 years ago

bc7b200f

Add ctr_mbl_feed_30x model

xuzhao9 committed 3 years ago

1063f311

fix explanations for PCIERX/PCIETX (#1043)

FindHao committed 3 years ago

8c50b668

update transformers to 4.20.1 (#1025)

FindHao committed 3 years ago

2e0ff47e

Support DDP in the core model set (#1031)

xuzhao9 committed 3 years ago

b6b19a9f

Add userbenchmark workflow (#1013)

xuzhao9 committed 3 years ago

86638bed

Add extra_info to test_bench.py reports (#1032)

xwang233 committed 3 years ago

cb019e82

fix help message of test_bench.py (#1021)

seastar105 committed 3 years ago

51445402

Fix the run.py results_summary for MPS. (#1020)

kulinseth committed 3 years ago

0528b0b9

Improve the correctness checking code and fix non-deterministic models (#997)

xuzhao9 committed 3 years ago

30981d92

Remove optimized_for_inference (#1012)

xuzhao9 committed 3 years ago

17222444

Add `--stress` option to `run.py` (#1002)

FindHao committed 3 years ago

1e95fdb6

Fix slow tests on CPU (#1003)

xuzhao9 committed 3 years ago

e0766561

Add detection2_fcos_r_50_fpn and support resizing inputs (#1001)

xuzhao9 committed 3 years ago

070d0fa8

Fix timm-efficientdet output (#998)

xuzhao9 committed 3 years ago

7bef86ea

Add torchdynamo cudagraph test (#993)

xuzhao9 committed 3 years ago

9a14a23d

reorganize imports to make model_analyzer more independent. (#1000)

FindHao committed 3 years ago

70ea2935

Add more GPU metrics to DCGM monitor (#989)

FindHao committed 3 years ago

7f1f4d2b

Add Dockerfile for building torchbench docker (#986)

xuzhao9 committed 3 years ago

023c5972

Add support for Super_SloMo fp16 (#987)

eellison committed 3 years ago

e0475800

Add deepspeed support (#959)

wconstab committed 3 years ago

acfd26ed

Update correctness checking code for fx2trt (#981)

xuzhao9 committed 3 years ago

191fb74d

Remove tools directory after checking it is in the expected location. (#979)

xuzhao9 committed 3 years ago

a599b49b

Newer Older