diff --git a/thunder/tests/test_inplace_functionalization.py b/thunder/tests/test_inplace_functionalization.py index a1f24f3ef6..d693becfc4 100644 --- a/thunder/tests/test_inplace_functionalization.py +++ b/thunder/tests/test_inplace_functionalization.py @@ -463,10 +463,12 @@ def f(a, b): def test_no_self_repeat_in_subsymbols(executor, device, _): def f(a, b, c): + a.add_(b, alpha=c) return a.add_(b, alpha=c) def functional_f(a, b, c): - return a.add(b, alpha=c) + d = a.add(b, alpha=c) + return d.add(b, alpha=c) a = make_tensor((2, 2), device=device, dtype=torch.float32) b = make_tensor((2, 2), device=device, dtype=torch.float32)