Hongxin Liu 4b3240cb59 [booster] add low level zero plugin (#3594) ... * [booster] add low level zero plugin * [booster] fix gemini plugin test * [booster] fix precision * [booster] add low level zero plugin test * [test] fix booster plugin test oom * [test] fix booster plugin test oom * [test] fix googlenet and inception output trans * [test] fix diffuser clip vision model * [test] fix torchaudio_wav2vec2_base * [test] fix low level zero plugin test		2023-04-26 14:37:25 +08:00
..
__init__.py	[booster] add low level zero plugin (#3594)	2023-04-26 14:37:25 +08:00
gemini_plugin.py	[misc] add verbose arg for zero and op builder (#3552)	2023-04-17 11:25:35 +08:00
low_level_zero_plugin.py	[booster] add low level zero plugin (#3594)	2023-04-26 14:37:25 +08:00
plugin_base.py	[booster] implemented the torch ddd + resnet example (#3232)	2023-03-27 10:24:14 +08:00
torch_ddp_plugin.py	[booster] fixed the torch ddp plugin with the new checkpoint api (#3442)	2023-04-06 09:43:51 +08:00