Skip to content

Commit

Permalink
adding brad runner with variable clients (run_variable_clients.py)
Browse files Browse the repository at this point in the history
  • Loading branch information
wuziniu committed Mar 19, 2024
1 parent 508ac5f commit 6917d7e
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions workloads/IMDB_extended/run_variable_clients.py
Original file line number Diff line number Diff line change
Expand Up @@ -414,15 +414,15 @@ class PauseController:
def __init__(
self,
total_num_clients: int,
pause_semaphore: mp.Semaphore, # type: ignore
resume_semaphore: mp.Semaphore, # type: ignore
pause_semaphore: List[mp.Semaphore], # type: ignore
resume_semaphore: List[mp.Semaphore], # type: ignore
):
self.total_num_clients = total_num_clients
self.pause_semaphore = pause_semaphore
self.resume_semaphore = resume_semaphore
self.paused_clients = []
self.running_clients = list(range(total_num_clients))
self.num_running_clients = total_num_clients
self.paused_clients: List[int] = []
self.running_clients: List[int] = list(range(total_num_clients))
self.num_running_clients: int = total_num_clients

def adjust_num_running_clients(
self, num_clients: int, verbose: bool = True
Expand Down

0 comments on commit 6917d7e

Please sign in to comment.