ColossalAI/colossalai/zero/gemini/chunk
Haze188 d9dddf574f
[Gemini] Use async stream to prefetch and h2d data moving (#5781)
* use async stream to prefetch and h2d data moving

* Remove redundant code
2024-06-12 15:48:52 +08:00
..
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
chunk.py [gemini] optimize reduce scatter d2h copy (#5760) 2024-06-05 14:23:13 +08:00
manager.py [Gemini] Use async stream to prefetch and h2d data moving (#5781) 2024-06-12 15:48:52 +08:00
search_utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
utils.py [Gemini] Use async stream to prefetch and h2d data moving (#5781) 2024-06-12 15:48:52 +08:00