search

BaguaSys / bagua-net

High performance NCCL plugin for Bagua.
https://bagua-tutorials.kwai-seattle.com/
MIT License
15 stars 4 forks source link

Add bagua-net support to bagua-core #6

Closed NOBLES5E closed 3 years ago

NOBLES5E commented 3 years ago

We can do this in a similar way as we add nccl to bagua-core: https://github.com/BaguaSys/bagua-core/blob/master/setup.py and https://github.com/BaguaSys/bagua-core/blob/master/python/bagua_core/_environment.py

We can control whether to enable bagua-net with the environment variable ENABLE_BAGUA_NET.

shjwudp commented 3 years ago

https://github.com/BaguaSys/bagua-core/pull/89