pytorch

4716fa24 - Improve dynamo support for autograd.Function (#98158)

Improve dynamo support for autograd.Function (#98158) Pull Request resolved: https://github.com/pytorch/pytorch/pull/98158 Approved by: https://github.com/yanboliang, https://github.com/anijain2305