Skip to content

Commit 639793d

Browse files
committed
fix for failing test
Signed-off-by: Lucas Liebenwein <[email protected]>
1 parent 08367d9 commit 639793d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unittest/_torch/auto_deploy/unit/singlegpu/compile/test_torch_opt.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ def __init__(self, base_model):
1919
def forward(self, x0, x1=None, x2=None):
2020
out = self.base_model(x0)
2121
if x1 is not None:
22-
out += self.base_model(x1)
22+
out = out + self.base_model(x1)
2323
if x2 is not None:
24-
out += self.base_model(x2)
24+
out = out + self.base_model(x2)
2525
return out
2626

2727

0 commit comments

Comments
 (0)