From ca73fc3b3d421c7f99bc87e58e4c30dff2d68e16 Mon Sep 17 00:00:00 2001 From: Sammy Kumar Date: Sat, 29 Nov 2025 09:00:20 -0600 Subject: [PATCH] fix(profiler_core.py): update OOM event to calculate execution time from start and end times for better performance tracking --- nodes/utils/resource_monitor/profiler_core.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/nodes/utils/resource_monitor/profiler_core.py b/nodes/utils/resource_monitor/profiler_core.py index 2b5f883..0c29a90 100644 --- a/nodes/utils/resource_monitor/profiler_core.py +++ b/nodes/utils/resource_monitor/profiler_core.py @@ -854,9 +854,14 @@ class ProfilerManager: node_type = oom_node.node_type # Create OOM event + # Calculate execution time from start/end + execution_time = None + if oom_node.start_time and oom_node.end_time: + execution_time = (oom_node.end_time - oom_node.start_time) * 1000 # Convert to ms + oom_event = { 'timestamp': datetime.now().isoformat(), - 'workflow_id': workflow_dict['workflowId'], + 'prompt_id': workflow_dict.get('promptId', 'unknown'), 'node_id': profile.oom_node_id, 'node_type': node_type, 'vram_free_before': oom_node.vram_free_before, @@ -865,7 +870,7 @@ class ProfilerManager: 'vram_peak': oom_node.vram_peak, 'models_at_oom': oom_node.models_at_oom, 'error_message': oom_node.oom_error, - 'execution_time': oom_node.execution_time + 'execution_time': execution_time } # Add to OOM history