Separating result processor out from profiler.py without changing the…

… behaviors of current profile.py
microsoft · Jan 6, 2025 · 0509b70 · 0509b70
1 parent c117213
commit 0509b70
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnxruntime/python/tools/transformers/profile_result_processor.py b/onnxruntime/python/tools/transformers/profile_result_processor.py
@@ -336,6 +336,7 @@ def group_node_results(sess_time):
 
     return lines
 
+
 def process_results(profile_file, args):
     profile_records = load_profile_json(profile_file)
 
@@ -347,6 +348,7 @@ def process_results(profile_file, args):
 
     return lines
 
+
 if __name__ == "__main__":
     arguments = parse_arguments()
     print("Arguments", arguments)
@@ -360,4 +362,4 @@ def process_results(profile_file, args):
     results = process_results(profile_file, arguments)
 
     for line in results:
-        print(line)
+        print(line)