ttnn-visualizer 0.55.0__py3-none-any.whl → 0.57.0__py3-none-any.whl

ttnn_visualizer/models.py

@@ -154,6 +154,26 @@ class StackTrace(SerializeableDataclass):
     stack_trace: str
 
 
+@dataclasses.dataclass
+class ErrorRecord(SerializeableDataclass):
+    operation_id: int
+    operation_name: str
+    error_type: str
+    error_message: str
+    stack_trace: str
+    timestamp: str
+
+    def to_nested_dict(self) -> dict:
+        """
+        Returns a dictionary representation without operation_id and operation_name.
+        Use this when the error is nested under an operation to avoid redundancy.
+        """
+        result = self.to_dict()
+        result.pop("operation_id", None)
+        result.pop("operation_name", None)
+        return result
+
+
 # Non Data Models
 
 

ttnn_visualizer/queries.py

@@ -12,6 +12,7 @@ from ttnn_visualizer.models import (
     BufferPage,
     Device,
     DeviceOperation,
+    ErrorRecord,
     InputTensor,
     Instance,
     Operation,
@@ -156,6 +157,13 @@ class DatabaseQueries:
             operation_id, stack_trace = row
             yield StackTrace(operation_id, stack_trace=stack_trace)
 
+    def query_error_records(
+        self, filters: Optional[Dict[str, Any]] = None
+    ) -> Generator[ErrorRecord, None, None]:
+        rows = self._query_table("errors", filters)
+        for row in rows:
+            yield ErrorRecord(*row)
+
     def query_tensor_comparisons(
         self, local: bool = True, filters: Optional[Dict[str, Any]] = None
     ) -> Generator[TensorComparisonRecord, None, None]:

ttnn_visualizer/serializers.py

@@ -19,6 +19,7 @@ def serialize_operations(
     devices,
     producers_consumers,
     device_operations,
+    error_records=None,
 ):
     tensors_dict = {t.tensor_id: t for t in tensors}
     device_operations_dict = {
@@ -29,6 +30,11 @@ def serialize_operations(
 
     stack_traces_dict = {st.operation_id: st.stack_trace for st in stack_traces}
 
+    errors_dict = {}
+    if error_records:
+        for error in error_records:
+            errors_dict[error.operation_id] = error.to_nested_dict()
+
     arguments_dict = defaultdict(list)
     for argument in operation_arguments:
         arguments_dict[argument.operation_id].append(argument)
@@ -49,6 +55,8 @@ def serialize_operations(
         )
         id = operation_data.pop("operation_id", None)
 
+        error_data = errors_dict.get(operation.operation_id)
+
         results.append(
             {
                 **operation_data,
@@ -58,6 +66,7 @@ def serialize_operations(
                 "arguments": arguments,
                 "inputs": inputs,
                 "outputs": outputs,
+                "error": error_data,
             }
         )
     return results
@@ -144,6 +153,7 @@ def serialize_operation(
     devices,
     producers_consumers,
     device_operations,
+    error_record=None,
 ):
     tensors_dict = {t.tensor_id: t for t in tensors}
     comparisons = comparisons_by_tensor_id(
@@ -176,6 +186,9 @@ def serialize_operation(
             device_operations_data = do.captured_graph
             break
 
+    # Convert error record to nested dict if it exists (excludes operation_id and operation_name)
+    error_data = error_record.to_nested_dict() if error_record else None
+
     return {
         **operation_data,
         "id": id,
@@ -186,6 +199,7 @@ def serialize_operation(
         "arguments": arguments_data,
         "inputs": inputs_data or [],
         "outputs": outputs_data or [],
+        "error": error_data,
     }
 
 

ttnn_visualizer/static/assets/{allPaths-BmK4MbG-.js → allPaths-ofNxJ79X.js}

@@ -1 +1 @@
-import{I as s}from"./index-CnPrfHYh.js";import{I as r}from"./index-Cnc1EkDo.js";import{p as n,I as c}from"./index-BvcHTS9m.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};
+import{I as s}from"./index-CnPrfHYh.js";import{I as r}from"./index-Cnc1EkDo.js";import{p as n,I as c}from"./index-Sc99nrOV.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};

ttnn_visualizer/static/assets/allPathsLoader-CN0GIv3K.js

@@ -0,0 +1,2 @@
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-ofNxJ79X.js","assets/index-CnPrfHYh.js","assets/index-Cnc1EkDo.js","assets/index-Sc99nrOV.js","assets/index-BO2e1Ec3.css"])))=>i.map(i=>d[i]);
+import{_ as e}from"./index-Sc99nrOV.js";const s=async(t,a)=>{const{getIconPaths:o}=await e(async()=>{const{getIconPaths:r}=await import("./allPaths-ofNxJ79X.js");return{getIconPaths:r}},__vite__mapDeps([0,1,2,3,4]));return o(t,a)};export{s as allPathsLoader};