Revert "Optimize tensor operation in neighbor_allreduce."

Bluefog-Lib · Mar 6, 2024 · 6e6f8c2 · 6e6f8c2
1 parent dd1344b
commit 6e6f8c2
Show file tree

Hide file tree

Showing 22 changed files with 51 additions and 1,632 deletions.
diff --git a/.gitignore b/.gitignore
@@ -129,8 +129,7 @@ data/
 **/checkpoint/
 experimental/
 
-# VSCode and JetBrains
-.idea/
+# VSCode
 .vscode/
 
 .DS_Store
@@ -141,13 +140,3 @@ examples/.profile/*
 # timeline log file
 .log/*
 .profile/
-
-# output
-img/*
-wandb/*
-output/*
-output*
-
-# script
-run**.sh
-scripts/
diff --git a/bluefoglite/common/__init__.py b/bluefoglite/common/__init__.py
@@ -1 +0,0 @@
-from bluefoglite.common import topology

diff --git a/bluefoglite/common/handle_manager.py b/bluefoglite/common/handle_manager.py
@@ -41,7 +41,6 @@ class EventStatus:
 
 DONE_EVENT = EventStatus(status=EventStatusEnum.DONE, err="")
 
-
 # This should be a singleton class
 class HandleManager:
     __instance: Optional["HandleManager"] = None

diff --git a/bluefoglite/common/optimizers.py b/bluefoglite/common/optimizers.py
diff --git a/bluefoglite/common/tcp/agent.py b/bluefoglite/common/tcp/agent.py
@@ -21,7 +21,6 @@
 from bluefoglite.common.tcp.pair import Pair, SocketFullAddress, TAddress
 from bluefoglite.common.logger import Logger
 
-
 # One agent can contain multiple Contexts.
 # Each Context should represent entire communication group like  (comm in MPI)
 # In each Context, it contains multiple Pairs, i.e. socket pair, talking to other neighbor.

diff --git a/bluefoglite/common/tcp/message_pb2.py b/bluefoglite/common/tcp/message_pb2.py
diff --git a/bluefoglite/common/tcp/pair.py b/bluefoglite/common/tcp/pair.py
@@ -47,7 +47,6 @@
 # AF_NETLINK, AF_TIPC) or strings (AF_UNIX).
 TAddress = Union[Tuple[Any, ...], str]
 
-
 # Socket related constant nomenclature follows:
 # AddressFamily startswith 'AF_'
 # SocketKind startswith 'SOCK_'