ColossalAI/colossalai/communication
アマデウス 6302069c0e
[model checkpoint] updated communication ops for cpu tensors (#590)
2022-04-01 16:52:20 +08:00
..
__init__.py Added profiler communication operations 2022-03-11 15:50:28 +08:00
collective.py [model checkpoint] updated communication ops for cpu tensors (#590) 2022-04-01 16:52:20 +08:00
p2p.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
ring.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
utils.py html refactor (#555) 2022-03-31 11:36:56 +08:00