apex icon indicating copy to clipboard operation
apex copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA

potential improvement in p2p communication

Open crcrpar opened this issue 3 years ago • 1 comments

https://github.com/NVIDIA/apex/blob/a0f5f3ac0f6bf39feee6e60eee66ec873dc299ab/apex/transformer/pipeline_parallel/p2p_communication.py#L271 might be able to be removed after confirming https://github.com/pytorch/pytorch/pull/82450

crcrpar avatar Aug 10 '22 20:08 crcrpar

in apex, the version check against NGC PyTorch containers (e.g. 22.08) and the released PyTorch (e.g. 1.12.1) would be sufficient

crcrpar avatar Aug 24 '22 16:08 crcrpar