Track node mutations (#152389)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/152389
Approved by: https://github.com/anijain2305
Reviewed By: huydhn
Differential Revision: D74531495
fbshipit-source-id: b8cc3f898efb29754cf34112b9b361b03a9caf7c
