pytorch

f4858fa8 - 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