Skip to content

Commit 5e3fcbb

Browse files
zhxchen17facebook-github-bot
authored andcommitted
The number of inputs given to exported program should be 2 not 3.
Summary: as title. The forward method only takes 2 inputs but we give 3 inputs to it during testing. Reviewed By: angelayi Differential Revision: D49501277 fbshipit-source-id: b0f809b73a455d7813d191a31c23cf3e997d2ef5
1 parent 175ca79 commit 5e3fcbb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/program/test/test_program.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -284,7 +284,7 @@ def test_edge_to_backend_selective(self):
284284

285285
forward_program = delegate_manager.exported_program("forward")
286286
self.assertEqual(
287-
forward_program(torch.ones(1), torch.ones(1), torch.ones(1)),
287+
forward_program(torch.ones(1), torch.ones(1)),
288288
torch.ones(1) + 1, # x * y + x
289289
)
290290

0 commit comments

Comments
 (0)