botbw 3f7e3131d9 [gemini] optimize reduce scatter d2h copy (#5760) ... * [gemini] optimize reduce scatter d2h copy * [fix] fix missing reduce variable * [refactor] remove legacy async reduce scatter code * [gemini] missing sync * Revert "[refactor] remove legacy async reduce scatter code" This reverts commit 58ad76d466. * [gemini] further optimize with async all reduce * [fix] pass flag from manager to chunk		2024-06-05 14:23:13 +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] optimize reduce scatter d2h copy (#5760)	2024-06-05 14:23:13 +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]remove registered gradients hooks (#5696)	2024-05-09 10:29:49 +08:00