mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-19 09:04:50 +00:00
4b3240cb59
* [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
7 lines
249 B
Python
7 lines
249 B
Python
from .gemini_plugin import GeminiPlugin
|
|
from .low_level_zero_plugin import LowLevelZeroPlugin
|
|
from .plugin_base import Plugin
|
|
from .torch_ddp_plugin import TorchDDPPlugin
|
|
|
|
__all__ = ['Plugin', 'TorchDDPPlugin', 'GeminiPlugin', 'LowLevelZeroPlugin']
|