Skip to content

Commit 104205d

Browse files
committed
removed debug prints
1 parent 4e4c47b commit 104205d

File tree

3 files changed

+1
-3
lines changed

3 files changed

+1
-3
lines changed

kernel_tuner/runners/parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,7 @@ def _check_gpus_equals(self):
211211
for env in environments:
212212
gpu_types.append(env["device_name"])
213213
if len(set(gpu_types)) == 1:
214-
print(f"DEBUG: Running on {len(gpu_types)} {gpu_types[0]}", file=sys.stderr)
214+
print(f"Running on {len(gpu_types)} {gpu_types[0]}", file=sys.stderr)
215215
return True
216216
else:
217217
return False

kernel_tuner/runners/ray/remote_actor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,6 @@ def init_runner(self):
6868
self.iterations, self.observers, cache_manager=self.cache_manager, dev=self.dev)
6969

7070
def _reinitialize_observers(self, observers_type_and_arguments):
71-
print("DEBUG: reinit observers called", file=sys.stderr)
7271
# observers can't be pickled to the actor so we need to re-initialize them
7372
self.observers = []
7473
for (observer, arguments) in observers_type_and_arguments:

kernel_tuner/util.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1288,7 +1288,6 @@ def get_num_devices(simulation_mode=False):
12881288
num_devices = round(resources.get("CPU") * 0.8)
12891289
else:
12901290
num_devices = resources.get("GPU")
1291-
print(f"DEBUG: {num_devices} Ray devices detected", file=sys.stderr)
12921291
return int(num_devices)
12931292

12941293
def get_gpu_id(lang):

0 commit comments

Comments
 (0)