Guilherme Leobas 491c2b4665 Let torch dynamo inline torch.func.grad (#118407) ...

When dynamo sees torch.func.grad, it tries to inline all frames related
to.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118407
Approved by: https://github.com/zou3519

2024-02-28 20:05:00 +00:00

..

__init__.py

Let torch dynamo inline torch.func.grad (#118407)

2024-02-28 20:05:00 +00:00