Making large AI models cheaper, faster and more accessible
[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 58ad76d4665032bbe548d066116d1c572ce98979. * [gemini] further optimize with async all reduce * [fix] pass flag from manager to chunk
B
botbw committed
3f7e3131d9f0fffbbd62bf7890c8cc94e8992a40
Parent: 10a19e2
Committed by GitHub <noreply@github.com>
on 6/5/2024, 6:23:13 AM