Skip to content

Commit 8067b1c

Browse files
committed
Fix lint
1 parent 41f1a52 commit 8067b1c

File tree

1 file changed

+23
-3
lines changed

1 file changed

+23
-3
lines changed

python/tvm/contrib/debugger/debug_executor.py

Lines changed: 23 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,15 @@ def _run_per_layer(self):
223223
output_tensors.append(self._get_node_output(i, j))
224224
self.debug_datum.update_output_tensors(output_tensors)
225225

226-
def _run_debug(self, number, repeat, min_repeat_ms, max_repeat_num, cooldown_interval_ms, repeats_to_cooldown):
226+
def _run_debug(
227+
self,
228+
number,
229+
repeat,
230+
min_repeat_ms,
231+
max_repeat_num,
232+
cooldown_interval_ms,
233+
repeats_to_cooldown,
234+
):
227235
"""Execute the node specified with index will be executed.
228236
Each debug output will be copied to the buffer
229237
Time consumed for each execution will be set as debug output.
@@ -335,7 +343,13 @@ def run(
335343
self.debug_datum.display_debug_result()
336344

337345
def run_individual(
338-
self, number, repeat=1, min_repeat_ms=0, max_repeat_num=100, cooldown_interval_ms=0, repeats_to_cooldown=1
346+
self,
347+
number,
348+
repeat=1,
349+
min_repeat_ms=0,
350+
max_repeat_num=100,
351+
cooldown_interval_ms=0,
352+
repeats_to_cooldown=1,
339353
):
340354
"""Run each operation in the graph and get the time per op for all ops.
341355
@@ -444,7 +458,13 @@ def run_individual_node(
444458
"""
445459
# Results are returned as serialized strings which we deserialize
446460
res = self._run_individual_node(
447-
index, number, repeat, min_repeat_ms, max_repeat_num, cooldown_interval_ms, repeats_to_cooldown
461+
index,
462+
number,
463+
repeat,
464+
min_repeat_ms,
465+
max_repeat_num,
466+
cooldown_interval_ms,
467+
repeats_to_cooldown,
448468
)
449469
fmt = "@" + ("d" * repeat)
450470
results = struct.unpack(fmt, res)

0 commit comments

Comments
 (0)