Skip to content

Commit b88b09c

Browse files
authored
Arm backend: Add missing attribute in VisualizePass (#14847)
VisualizePass was missing _passes_required_after and could therefore not be initialized. Define this attribute to fix the problem. Signed-off-by: Martin Lindström <[email protected]>
1 parent a41cdef commit b88b09c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

backends/arm/_passes/_debug_passes.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
# This source code is licensed under the BSD-style license found in the
44
# LICENSE file in the root directory of this source tree.
55

6+
from typing import Set, Type
7+
68
import torch
79
from executorch.devtools.visualization.visualization_utils import visualize_graph
810
from executorch.exir import ExportedProgram
@@ -14,6 +16,8 @@ class VisualizePass(ExportPass):
1416
This pass visualizes the graph at the point of insertion in the pass manager
1517
"""
1618

19+
_passes_required_after: Set[Type[ExportPass]] = set()
20+
1721
def __init__(self, exported_program: ExportedProgram) -> None:
1822
super().__init__()
1923
self.exported_program = exported_program

0 commit comments

Comments
 (0)