pytorch/benchmark

Pull Requests Commits

Remove two more models (#812)

xuzhao9 committed 3 years ago

Verified 6077a453

Remove numba dependencies (#809)

xuzhao9 committed 3 years ago

Verified f4516d4b

Remove numpy from requirements.txt

desertfire committed 3 years ago

eb5fff26

Merge pull request #803 from pytorch/ltc_noopt_merge

desertfire committed 3 years ago

Verified 8843c8a3

Change CI script to use noopt mode for lazy_bench

wconstab committed 3 years ago

0d4475db

Move the computation of self.example_outputs to be training only

desertfire committed 3 years ago

bac024cf

Pass in a `test` option when calling Model

desertfire committed 3 years ago

d824fe7a

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

xuzhao9 committed 3 years ago

e811b50d

Merge branch 'main' into ltc_merge

desertfire committed 3 years ago

9069908f

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

Older