pytorch/benchmark

Pull Requests Commits

Fix bisection workflow broken by vision and ffmpeg upgrade (#797)

xuzhao9 committed 3 years ago

2a8528f9

Implement unicode_csv_reader because it was removed in upstream. (#795)

xuzhao9 committed 3 years ago

59b1758f

Enable CPU test on speech_transformer (for accuracy test) (#791)

xuzhao9 committed 3 years ago

06beb921

Add CPU implementation to pytorch_CycleGAN_and_pix2pix (#790)

xuzhao9 committed 3 years ago

0f641a08

Remove model version lock and test on Python 3.9 (#787)

xuzhao9 committed 3 years ago

7a6ab934

Fixes Background_Matting get_module interface (#785)

xuzhao9 committed 3 years ago

dc94612d

Enable fp16 by default for hf models (#782)

xuzhao9 committed 3 years ago

536969ca

Use fuser contextmanager (#784)

xuzhao9 committed 3 years ago

93a4a7fb

Remove "NotImplemented" flags if the test is limited by hardware capacity (#781)

xuzhao9 committed 3 years ago

f2c83718

Enable fx2trt on Huggingface models (#778)

xuzhao9 committed 3 years ago

ee89c449

Add correctness CI. (#779)

xuzhao9 committed 3 years ago

99086b33

Add fp16 amp mode support for all models (#776)

xuzhao9 committed 3 years ago

cc4d824d

Add the hf_bert E2E benchmark (#771)

xuzhao9 committed 3 years ago

3d672dde

Factor the core lowering component into a pass function (#777)

kflu committed 3 years ago

c714a6e2

Fix a env check bug for ondemand ci (#770)

xuzhao9 committed 3 years ago

4bf35cf7

Cleanup the benchmarking code for timm models (#772)

xuzhao9 committed 3 years ago

f56637b2

Add the actual TensorRT correctness testing code (#763)

xuzhao9 committed 3 years ago

b904ffdb

First PR to add the correctness checking code to eval tests (#762)

xuzhao9 committed 3 years ago

ca731c26

Move the jit script code to __post_init__ (#761)

xuzhao9 committed 3 years ago

3173d623

Add pygame with gym requirement (#766)

kit1980 committed 3 years ago

0743d73f

Enable fp16 for torchvision models inference CUDA tests by default (#758)

xuzhao9 committed 3 years ago

2ee0fd06

Fix a few problems discovered when running the workflow. (#756)

xuzhao9 committed 3 years ago

d6686732

Enable torch_trt and handle subprocess segfault in subprocess_worker (#754)

xuzhao9 committed 3 years ago

0c9e3002

Revert #736 with maml and maml_omiglot (#752)

xuzhao9 committed 3 years ago

14d06881

Add nvfuser support for torchvision models (#744)

xuzhao9 committed 3 years ago

775b254a

Enable fx2trt and torch_tensorrt support across all TB models (#743)

xuzhao9 committed 3 years ago

e0deda39

Adds the sweep runner script (#742)

xuzhao9 committed 3 years ago

69723a96

Move fx2trt out of PyTorch core (#72499)

Yinghai Lu committed 3 years ago

f91d1939

Remove the maml models as they have poor quality and should be replaced. (#736)

xuzhao9 committed 3 years ago

907a4eba

Add a new "test" argument to models (#735)

xuzhao9 committed 3 years ago

b93c9ebc

Newer Older