ColossalAI/colossalai/gemini/chunk
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
..
__init__.py [zero] add chunk init function for users (#1729) 2022-10-18 16:31:22 +08:00
chunk.py [zero] add constant placement policy (#1705) 2022-10-14 17:53:16 +08:00
manager.py [zero] add constant placement policy (#1705) 2022-10-14 17:53:16 +08:00
search_utils.py [zero] add chunk init function for users (#1729) 2022-10-18 16:31:22 +08:00
utils.py [zero] add chunk init function for users (#1729) 2022-10-18 16:31:22 +08:00