ColossalAI/colossalai/booster/plugin
Xuanlei Zhao 3acbf6d496
[npu] add npu support for hybrid plugin and llama (#5090)
* llama 3d

* update

* fix autocast
2023-11-22 19:23:21 +08:00
..
__init__.py
dp_plugin_base.py
gemini_plugin.py [format] applied code formatting on changed files in pull request 5067 (#5072) 2023-11-20 19:46:43 +08:00
hybrid_parallel_plugin.py [npu] add npu support for hybrid plugin and llama (#5090) 2023-11-22 19:23:21 +08:00
low_level_zero_plugin.py [npu] add npu support for gemini and zero (#5067) 2023-11-20 16:12:41 +08:00
moe_hybrid_parallel_plugin.py
plugin_base.py
pp_plugin_base.py
torch_ddp_plugin.py
torch_fsdp_plugin.py