Add a bit perf more logging

Author: ilevkivskyi

mypy/build.py

@@ -1173,13 +1173,15 @@ def stats_summary(self) -> Mapping[str, object]:
 
     def broadcast(self, message: bytes) -> None:
         """Broadcast same message to all workers in parallel."""
+        t0 = time.time()
         threads = []
         for worker in self.workers:
             thread = Thread(target=worker.conn.write_bytes, args=(message,))
             thread.start()
             threads.append(thread)
         for thread in threads:
             thread.join()
+        self.add_stats(broadcast_time=time.time() - t0)
 
     def wait_ack(self) -> None:
         """Wait for an ack from all workers."""
@@ -1207,6 +1209,7 @@ def submit_to_workers(self, graph: Graph, sccs: list[SCC] | None = None) -> None
                 for mod_id in scc.mod_ids
                 if (path := graph[mod_id].xpath) in self.errors.recorded
             }
+            t0 = time.time()
             send(
                 self.workers[idx].conn,
                 SccRequestMessage(
@@ -1224,6 +1227,7 @@ def submit_to_workers(self, graph: Graph, sccs: list[SCC] | None = None) -> None
                     },
                 ),
             )
+            self.add_stats(scc_send_time=time.time() - t0)
 
     def wait_for_done(
         self, graph: Graph

mypy/build_worker/worker.py

@@ -197,7 +197,9 @@ def serve(server: IPCServer, ctx: ServerContext) -> None:
         except CompileError as blocker:
             send(server, SccResponseMessage(scc_id=scc_id, blocker=blocker))
         else:
+            t1 = time.time()
             send(server, SccResponseMessage(scc_id=scc_id, result=result))
+            manager.add_stats(scc_send_time=time.time() - t1)
         manager.add_stats(total_process_stale_time=time.time() - t0, stale_sccs_processed=1)