pytorch/benchmark

Pull Requests Commits

use NotImplementedError

wconstab committed 3 years ago

80bb8266

Merge branch 'main' into wconstab/dynamic

wconstab committed 3 years ago

Verified 10c32c16

construct DS examples inside getter to avoid always constructing them

wconstab committed 3 years ago

dacf4247

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

WIP got bert working

wconstab committed 3 years ago

0d3a576d

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

Fix train test and tools package in the detectron2 model set (#966)

xuzhao9 committed 3 years ago

2fd26259

Migrate the workflow to use the new torch-tensorrt package (#965)

xuzhao9 committed 3 years ago

94d51e17

Older