Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit fed20d2

Browse files
thorjohnsenthorjohnsen
authored
Merge pull request #1340 from NVIDIA/peer_memory
Peer memory halo exchange
2 parents d89f5e6 + 5698eee commit fed20d2

16 files changed

Lines changed: 3624 additions & 428 deletions

‎apex/contrib/bottleneck/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,2 @@
11
from .bottleneck import Bottleneck, SpatialBottleneck
2+
from .halo_exchangers import HaloExchangerNoComm, HaloExchangerAllGather, HaloExchangerSendRecv, HaloExchangerPeer

‎apex/contrib/bottleneck/bottleneck.py‎

Lines changed: 374 additions & 144 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)