Skip to content

Commit f2a7f46

Browse files
authored
Arm backend: Update arm_tester.py with generate_etrecord (#13370)
Seems 94f3880 updated run() with another parameter that needs to be handled. Signed-off-by: [email protected]
1 parent 910918e commit f2a7f46

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

backends/arm/test/tester/arm_tester.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,14 +171,17 @@ def dump_artifact(self, path_to_dump: Optional[str]):
171171
super().dump_artifact(path_to_dump)
172172
_dump_lowered_modules_artifact(path_to_dump, self.artifact, self.graph_module)
173173

174-
def run(self, artifact: ExportedProgram, inputs=None) -> None:
174+
def run(
175+
self, artifact: ExportedProgram, inputs=None, generate_etrecord: bool = False
176+
) -> None:
175177
artifact_to_run = copy.deepcopy(artifact)
176178
self.edge_dialect_program = to_edge_transform_and_lower(
177179
artifact_to_run,
178180
transform_passes=self.transform_passes,
179181
compile_config=self.edge_compile_conf,
180182
partitioner=self.partitioners,
181183
constant_methods=self.constant_methods,
184+
generate_etrecord=generate_etrecord,
182185
)
183186

184187

0 commit comments

Comments
 (0)