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-01-16 07:28:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f8aeecef46461ff574f51982d03310fa8c57888e
ColossalAI/colossalai/gemini
History
HELSON f69f9bf223 [zero] add chunk init function for users (#1729)
* add chunk manager init function

* fix unit tests

* add comment

* add flush=True
2022-10-18 16:31:22 +08:00
..
chunk
[zero] add chunk init function for users (#1729)
2022-10-18 16:31:22 +08:00
memory_tracer
[feature] A new ZeRO implementation (#1644)
2022-10-09 09:18:51 +08:00
ophooks
[hotfix] remove potiential circle import (#1307)
2022-07-14 13:44:26 +08:00
paramhooks
[hotfix] remove potiential circle import (#1307)
2022-07-14 13:44:26 +08:00
__init__.py
[feature] A new ZeRO implementation (#1644)
2022-10-09 09:18:51 +08:00
gemini_context.py
[hotfix] add deconstructor for stateful tensor (#848)
2022-04-24 15:03:04 +08:00
gemini_mgr.py
[zero] add constant placement policy (#1705)
2022-10-14 17:53:16 +08:00
placement_policy.py
[zero] add constant placement policy (#1705)
2022-10-14 17:53:16 +08:00
stateful_tensor_mgr.py
[gemini] accelerate adjust_layout() (#878)
2022-04-26 18:08:31 +08:00
stateful_tensor.py
[hotfix] add deconstructor for stateful tensor (#848)
2022-04-24 15:03:04 +08:00
tensor_placement_policy.py
[gemini] accelerate adjust_layout() (#878)
2022-04-26 18:08:31 +08:00
tensor_utils.py
[gemini] add GeminiMemoryManger (#832)
2022-04-24 13:08:48 +08:00
Powered by Gitea Version: 1.25.2 Page: 957ms Template: 51ms
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