Handle empty label after heuristics

ARM-software · Jan 10, 2025 · cb6e795 · cb6e795
1 parent 29800d2
commit cb6e795
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lglpy/timeline/data/processed_trace.py b/lglpy/timeline/data/processed_trace.py
@@ -158,11 +158,14 @@ def get_workload_name(self) -> str:
         Returns:
             Returns the label for use in the UI.
         '''
-        if not self.label_stack:
+        label = None
+        if self.label_stack:
+            label = self.get_label_name()
+
+        # Default label if no label or get_label_name heuristics stripped it
+        if not label:
             return GPUStageID.get_ui_name(self.stage)
 
-        label = self.get_label_name()
-        assert label
         return label
 
     def get_long_label(self) -> str: