| 97 | |
| 98 | @dataclass |
| 99 | class DebugSchema: |
| 100 | event_id: int |
| 101 | aten_info: ATenDebugSchema |
| 102 | tosa_info: Optional[TosaDebugSchema] |
| 103 | torch_info: TorchDebugSchema |
| 104 | |
| 105 | def to_dict(self) -> dict[str, Any]: |
| 106 | output = asdict(self) |
| 107 | |
| 108 | if self.tosa_info is None: |
| 109 | output.pop("tosa_info") |
| 110 | |
| 111 | return output |
| 112 | |
| 113 | |
| 114 | class DebugHook: |