pytorch/benchmark

Pull Requests Commits

Fix timm-efficientdet output (#998)

xuzhao9 committed 4 years ago

7bef86ea

Add torchdynamo cudagraph test (#993)

xuzhao9 committed 4 years ago

9a14a23d

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

FindHao committed 4 years ago

70ea2935

Add more GPU metrics to DCGM monitor (#989)

FindHao committed 4 years ago

7f1f4d2b

Add Dockerfile for building torchbench docker (#986)

xuzhao9 committed 4 years ago

023c5972

Add support for Super_SloMo fp16 (#987)

eellison committed 4 years ago

e0475800

Add deepspeed support (#959)

wconstab committed 4 years ago

acfd26ed

Update correctness checking code for fx2trt (#981)

xuzhao9 committed 4 years ago

191fb74d

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

xuzhao9 committed 4 years ago

a599b49b

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

xuzhao9 committed 4 years ago

2fd26259

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

xuzhao9 committed 4 years ago

94d51e17

Remove the pplbench beanmachine model (#968)

xuzhao9 committed 4 years ago

ca502991

Fix the te fuser setup (#963)

xuzhao9 committed 4 years ago

842466b5

Add detectron2 series of models (#955)

xuzhao9 committed 4 years ago

92c20c6a

add --cuda_only; fix some nvfuser flags (#960)

xwang233 committed 4 years ago

1b81105e

Add unittest-xml-reporting deps to fix the broken CI (#958)

xuzhao9 committed 4 years ago

52c067fc

Add userbenchmark metrics to the distributed benchmark (#954)

xuzhao9 committed 4 years ago

f30d537c

update regnet and detectron models (#951)

FindHao committed 4 years ago

96938549

Pin numpy version and mkl libraries (#944)

xuzhao9 committed 4 years ago

9bb0bd70

Swap fx2trt_oss to torch_tensorrt (#92)

Wei Wei committed 4 years ago

476ebd74

Support ctr_mbl_feed_30x_over model in TorchBench Internal

xuzhao9 committed 4 years ago

5aad9057

Benchmark aotautograd+nvfuser (#932)

xuzhao9 committed 4 years ago

45a845ea

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

xuzhao9 committed 4 years ago

51a70428

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

kulinseth committed 4 years ago

4cc1de9e

Add abtest script for userbenchmark PR testing (#936)

xuzhao9 committed 4 years ago

6200aa1b

Update the config to use a100 runner (#926)

xuzhao9 committed 4 years ago

d7ec7b2b

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

xuzhao9 committed 4 years ago

8300ff42

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

FindHao committed 4 years ago

b5e0b83c

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

xuzhao9 committed 4 years ago

8b2dc796

Pin the version of dill for attention_is_all_you_need_pytorch (#922)

xuzhao9 committed 4 years ago

eb672435

Newer Older