Revert the changes for elif branch

pytorch · May 8, 2023 · 5bdcbcc · 5bdcbcc
1 parent 61a8413
commit 5bdcbcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_dynamo/variables/torch.py b/torch/_dynamo/variables/torch.py
@@ -648,7 +648,7 @@ def check_any_unspec(x):
                         if name in tx.symbolic_locals:
                             tx.symbolic_locals[name] = tensor_variable.items[idx]
                 elif isinstance(tensor_variable, TensorVariable):
-                    assert isinstance(kwargs["out"], (TensorVariable, ListVariable))
+                    assert isinstance(kwargs["out"], TensorVariable)
                     name = tx.find_symbolic_locals_name(kwargs["out"])
                     if name in tx.symbolic_locals:
                         tx.symbolic_locals[name] = tensor_variable