Logo
Explore Help
Register Sign In
github/ColossalAI
1
0
Fork 0
You've already forked ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 23:22:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d151dcab740eaae784333c93d85100c3641bd115
ColossalAI/colossalai/shardformer/layer
History
Bin Jia 86d22581e4 [shardformer] Add overlap optional for HybridParallelPlugin (#4615)
* add optional overlap for plugin

* remove fixed todo
2023-09-05 11:52:23 +08:00
..
__init__.py
[shardformer] fix import
2023-08-15 23:25:14 +08:00
_operation.py
[shardformer] Add overlap optional for HybridParallelPlugin (#4615)
2023-09-05 11:52:23 +08:00
dropout.py
[shardformer] supported bloom model (#4098)
2023-07-04 16:05:01 +08:00
embedding.py
[shardformer] fix embedding
2023-08-15 23:25:14 +08:00
linear.py
[shardformer/fix overlap bug] fix overlap bug, add overlap as an option in shardco… (#4516)
2023-08-28 17:16:40 +08:00
loss.py
fix some typo colossalai/shardformer (#4160)
2023-07-04 17:53:39 +08:00
normalization.py
[shardformer] support inplace sharding (#4251)
2023-08-15 23:25:14 +08:00
parallel_module.py
[shardformer] support sharded checkpoint IO for models of HybridParallelPlugin (#4506)
2023-08-25 22:04:57 +08:00
qkv_fused_linear.py
[shardformer] Add overlap support for gpt2 (#4535)
2023-08-29 18:30:50 +08:00
utils.py
[misc] resolve code factor issues (#4433)
2023-08-15 23:25:14 +08:00
Powered by Gitea Version: 1.25.2 Page: 1545ms Template: 8ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API