Skip to content

Commit 7f1f0ed

Browse files
committed
[feat] support msprof-analyze, add analsys result
1 parent 9d43ef7 commit 7f1f0ed

File tree

8 files changed

+7930
-35
lines changed

8 files changed

+7930
-35
lines changed

applications/ColossalChat/coati/distributed/consumer.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
from .comm import ray_broadcast_tensor_dict
1919
from .utils import bind_batch, post_recv, unbind_batch
2020

21-
first_sleep=True
2221
class BaseConsumer:
2322
def __init__(
2423
self,
@@ -124,11 +123,6 @@ def loop(self) -> None:
124123
# receive data from producers
125124
for r in range(self.num_producers):
126125
print(f"[T{dist.get_rank()}] Recv data episode {episode} step {step} from {r}")
127-
global first_sleep
128-
if first_sleep:
129-
import time
130-
time.sleep(720)
131-
first_sleep=False
132126
self.buffer.extend(
133127
unbind_batch(
134128
ray_broadcast_tensor_dict(
99.9 KB
Binary file not shown.

0 commit comments

Comments
 (0)