pytorch/benchmark

Pull Requests Commits

wconstab committed 3 years ago

59914953

Add arg for running dist benchmarks locally

wconstab committed 3 years ago

1077e519

Add userbenchmark metrics to the distributed benchmark (#954)

xuzhao9 committed 3 years ago

f30d537c

update regnet and detectron models (#951)

FindHao committed 3 years ago

96938549

Pin numpy version and mkl libraries (#944)

xuzhao9 committed 3 years ago

9bb0bd70

Swap fx2trt_oss to torch_tensorrt (#92)

Wei Wei committed 3 years ago

476ebd74

Support ctr_mbl_feed_30x_over model in TorchBench Internal

xuzhao9 committed 3 years ago

5aad9057

Benchmark aotautograd+nvfuser (#932)

xuzhao9 committed 3 years ago

45a845ea

Fix a few bugs in the userbenchmark abtest script (#940)

xuzhao9 committed 3 years ago

51a70428

Add the MPS device for the runs. (#917)

kulinseth committed 3 years ago

4cc1de9e

Add abtest script for userbenchmark PR testing (#936)

xuzhao9 committed 3 years ago

6200aa1b

Update the config to use a100 runner (#926)

xuzhao9 committed 3 years ago

d7ec7b2b

Pre-download multi30k dataset for attention model (#937)

xuzhao9 committed 3 years ago

8300ff42

Add support for NVIDIA DCGM based FLOPs/sec calculation (#929)

FindHao committed 3 years ago

b5e0b83c

Fix a minor issue in the domain score calculation. (#908)

xuzhao9 committed 3 years ago

8b2dc796

Pin the version of dill for attention_is_all_you_need_pytorch (#922)

xuzhao9 committed 3 years ago

eb672435

Add amp+train support and channels_last support to timm models (#900)

xuzhao9 committed 3 years ago

5a152887

Add required example_inputs argument to prepare_fx and prepare_qat_fx (#77608)

jerryzh168 committed 3 years ago

6f0d2fbd

NVFuser userbenchmarks - use JSON output format (#920)

davidberard98 committed 3 years ago

ae802fdd

Increase the default batch size of alexnet (#919)

xuzhao9 committed 3 years ago

af847a96

Pin the omegaconf package version for xlmr (#918)

xuzhao9 committed 3 years ago

113887cb

Patch torchdynamo to workaround dynamo#159 (#874)

xuzhao9 committed 3 years ago

75ca4de4

Fix fp16 precision and cos sim correctness (#913)

xuzhao9 committed 3 years ago

b81d24d2

Fix nvfuser benchmarks (#915)

davidberard98 committed 3 years ago

59d12a58

Promote microbenchmarks to userbenchmark (#914)

xuzhao9 committed 3 years ago

211a0ef2

Improve correctness algorithm (#886)

xuzhao9 committed 3 years ago

4bbdc7b9

Add DistributedDataParallel Benchmark for Single-GPU, Multi-GPU, and Multi-Machine (#850)

mrshenli committed 3 years ago

d258f016

Pin fairseq version to 0.11.0 (#902)

xuzhao9 committed 3 years ago

2ae910d7

fix opacus nn.functional import (#426)

Natalia Gimelshein committed 3 years ago

6d209470

Update fambench_dlrm configs (#883)

xuzhao9 committed 3 years ago

f6c4d271

Older