Let torch dynamo inline torch.func.grad (#118407)
Summary:
When dynamo sees torch.func.grad, it tries to inline all frames related
to.
X-link: https://github.com/pytorch/pytorch/pull/118407
Approved by: https://github.com/zou3519
Reviewed By: kit1980
Differential Revision: D54343701
fbshipit-source-id: aa85062ebb7d1486f26f7d7fe11d8eb2462dc458
