Skip to content

Commit

Permalink
Remove prints
Browse files Browse the repository at this point in the history
  • Loading branch information
shuds13 committed May 23, 2023
1 parent 3940606 commit 089419f
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 14 deletions.
11 changes: 0 additions & 11 deletions libensemble/resources/worker_resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,10 +59,6 @@ def __init__(self, num_workers: int, resources: "GlobalResources") -> None: # n
self.gpu_rsets_free = self.total_num_gpu_rsets
self.nongpu_rsets_free = self.total_num_nongpu_rsets

# print(f"\n{self.total_num_rsets=}")
# print(f"{self.total_num_gpu_rsets=}")
# print(f"{self.total_num_nongpu_rsets=}\n")

# Useful for scheduling tasks with different sized groups (resource sets per node).
unique, counts = np.unique(self.rsets["group"], return_counts=True)
self.group_sizes = dict(zip(unique, counts))
Expand All @@ -85,9 +81,6 @@ def assign_rsets(self, rset_team, worker_id):
ResourceManagerException(
f"Error: Attempting to assign rsets {rset_team}" f" already assigned to workers: {rteam}"
)
# print(f"\nAfter assign: {self.rsets_free=}")
# print(f"After assign: {self.gpu_rsets_free=}")
# print(f"After assign: {self.nongpu_rsets_free=}\n")

def free_rsets(self, worker=None):
"""Free up assigned resource sets"""
Expand All @@ -102,10 +95,6 @@ def free_rsets(self, worker=None):
self.rsets_free += len(rsets_to_free)
self.gpu_rsets_free += np.count_nonzero(self.rsets["gpus"][rsets_to_free])
self.nongpu_rsets_free += np.count_nonzero(~self.rsets["gpus"][rsets_to_free])
# print(f"\nfreed up {len(rsets_to_free)} ")
# print(f"---After free: {self.rsets_free=}")
# print(f"---After free: {self.gpu_rsets_free=}")
# print(f"---After free: {self.nongpu_rsets_free=}\n")

@staticmethod
def get_index_list(
Expand Down
3 changes: 0 additions & 3 deletions libensemble/tools/alloc_support.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,6 @@ def sim_work(self, wid, H, H_fields, H_rows, persis_info, **libE_info):
"""
# Parse out resource_sets
self._update_rset_team(libE_info, wid, H=H, H_rows=H_rows)
# print(f"Sim: {libE_info=}")

H_fields = AllocSupport._check_H_fields(H_fields)
libE_info["H_rows"] = AllocSupport._check_H_rows(H_rows)
Expand Down Expand Up @@ -244,8 +243,6 @@ def gen_work(self, wid, H_fields, H_rows, persis_info, **libE_info):
"""
self._update_rset_team(libE_info, wid)

# print(f"Gen: {libE_info=}")

if not self.W[wid - 1]["persis_state"]:
AllocSupport.gen_counter += 1 # Count total gens
libE_info["gen_count"] = AllocSupport.gen_counter
Expand Down

0 comments on commit 089419f

Please sign in to comment.