@@ -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