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-11 14:43:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
38ea4ba1bdae4b04e90dc1cd5e71e3944c82f805
ColossalAI/colossalai/gemini
History
Zihao 38ea4ba1bd [Gemini] fix grad unreleased issue and param recovery issue (#2052)
2022-12-02 16:04:19 +08:00
..
chunk
[hotfix] fix zero's incompatibility with checkpoint in torch-1.12 (#1786)
2022-11-02 16:11:34 +08:00
memory_tracer
[Gemini] fix grad unreleased issue and param recovery issue (#2052)
2022-12-02 16:04:19 +08:00
ophooks
[Gemini] fix grad unreleased issue and param recovery issue (#2052)
2022-12-02 16:04:19 +08:00
paramhooks
[hotfix] remove potiential circle import (#1307)
2022-07-14 13:44:26 +08:00
__init__.py
[hotfix] polish chunk import (#1787)
2022-11-02 12:10:52 +08:00
gemini_context.py
[hotfix] add deconstructor for stateful tensor (#848)
2022-04-24 15:03:04 +08:00
gemini_mgr.py
[Gemini] polish memstats collector (#1962)
2022-11-16 15:45:57 +08:00
placement_policy.py
[Gemini] polish memstats collector (#1962)
2022-11-16 15:45:57 +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] free and allocate cuda memory by tensor.storage, add grad hook (#2040)
2022-11-30 15:57:45 +08:00
Powered by Gitea Version: 1.25.2 Page: 145ms Template: 6ms
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