There was an error while loading. Please reload this page.
1 parent 2ab752d commit 348272eCopy full SHA for 348272e
torch/_subclasses/fake_tensor.py
@@ -1633,9 +1633,7 @@ def _prep_args_for_hash(
1633
# Special case for AOT Dispatcher first pass, where the fake
1634
# tensor is called on the functional wrapper of the subgraph.
1635
result.append(hash(arg))
1636
- # functional wrapper is destroyed after fake tensor prop. We
1637
- # need to put the finalizer on the subgraph.
1638
- id_hashed_objects.append(arg.subgraph)
+ id_hashed_objects.append(arg)
1639
else:
1640
# It's important to capture the type of the arg since, e.g., 1 and 1.0
1641
# hash to the same value, but can produce different dtypes for the
0 commit comments