SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 36 Python

[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