Merge from the main branch #796
desertfire
merged
44 commits into
wconstab/ltc
from
ltc_merge
Enable CUDA graph train on torchvision models (#706)
e50b595b
Fix V2 bisection workflow bug. (#709)
5f707392
Fixed the workflow file and upload script. (#712)
aca0584e
Remove the unused waveglow requirements file (#716)
76fa3723
Allow running install.py from other directories (#719)
830e1654
Fix the installer script path (#721)
dc42262b
Use py38 and CUDA 11.3 as default in the nightly wheel script (#718)
88125e01
Fix a bug when there is no anomaly detected (#728)
0f239843
Add the test workflow. (#729)
67a68c9c
Silence printouts in fastNLP_Bert and vision_maskrcnn (#722)
a389101d
changed eval_bs to train_bs (#731)
453b22eb
Support torch_tensorrt for torchvision and timm models (#732)
6cf0ec0a
Add a new "test" argument to models (#735)
b93c9ebc
Remove the maml models as they have poor quality and should be replac…
907a4eba
Move fx2trt out of PyTorch core (#72499)
f91d1939
Adds the sweep runner script (#742)
69723a96
Enable fx2trt and torch_tensorrt support across all TB models (#743)
e0deda39
Add nvfuser support for torchvision models (#744)
775b254a
Revert #736 with maml and maml_omiglot (#752)
14d06881
Enable torch_trt and handle subprocess segfault in subprocess_worker …
0c9e3002
Fix a few problems discovered when running the workflow. (#756)
d6686732
Enable fp16 for torchvision models inference CUDA tests by default (#…
2ee0fd06
Add pygame with gym requirement (#766)
0743d73f
Move the jit script code to __post_init__ (#761)
3173d623
First PR to add the correctness checking code to eval tests (#762)
ca731c26
Add the actual TensorRT correctness testing code (#763)
b904ffdb
Cleanup the benchmarking code for timm models (#772)
f56637b2
Fix a env check bug for ondemand ci (#770)
4bf35cf7
Factor the core lowering component into a pass function (#777)
c714a6e2
Add the hf_bert E2E benchmark (#771)
3d672dde
Add fp16 amp mode support for all models (#776)
cc4d824d
Add correctness CI. (#779)
99086b33
Enable fx2trt on Huggingface models (#778)
ee89c449
Remove "NotImplemented" flags if the test is limited by hardware capa…
f2c83718
Use fuser contextmanager (#784)
93a4a7fb
Enable fp16 by default for hf models (#782)
536969ca
Fixes Background_Matting get_module interface (#785)
dc94612d
Remove model version lock and test on Python 3.9 (#787)
7a6ab934
Add CPU implementation to pytorch_CycleGAN_and_pix2pix (#790)
0f641a08
Enable CPU test on speech_transformer (for accuracy test) (#791)
06beb921
Merge branch 'main' into ltc_merge
9069908f
Implement unicode_csv_reader because it was removed in upstream. (#795)
e811b50d
Pass in a `test` option when calling Model
d824fe7a
Move the computation of self.example_outputs to be training only
bac024cf
wconstab
approved these changes
on 2022-03-16
Reviewers
Assignees
No one assigned
Labels
cla signed
Milestone
No milestone
Login to write a write a comment.
Login via GitHub