Skip to content

Commit 131d1f4

Browse files
Arm backend: Make PassPipeline test valid inputs (pytorch#15892)
Pass valid test_data to run_method_and_compare_outputs in PassPipeline. Signed-off-by: Oscar Andersson <[email protected]>
1 parent 17e4392 commit 131d1f4

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

backends/arm/test/tester/test_pipeline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -806,7 +806,10 @@ def __init__(
806806
self.add_stage(self.tester.check_count, ops_after_pass, suffix="after")
807807
if ops_not_after_pass:
808808
self.add_stage(self.tester.check_not, ops_not_after_pass, suffix="after")
809-
self.add_stage(self.tester.run_method_and_compare_outputs)
809+
self.add_stage(
810+
self.tester.run_method_and_compare_outputs,
811+
inputs=self.test_data,
812+
)
810813

811814
def run(self):
812815
with TosaLoweringContext(self.tosa_spec):

0 commit comments

Comments
 (0)