Guilherme Leobas 4eaa000acc Teach dynamo about torch.func.jvp (#119926) ... List of changes: - Replace JVP_NESTING by torch._C._functorch.maybe_current_level() - Remove all increment nesting functions from wrap_fx_proxy_cls - fwAD.make_dual receives the dual_level as keyword argument - Add jvp_increment_nesting, set_fwd_grad_enabled and dual_level context managers to dynamo Pull Request resolved: https://github.com/pytorch/pytorch/pull/119926 Approved by: https://github.com/zou3519		2024-03-22 20:25:47 +00:00
..
init.cpp	Teach dynamo about torch.func.jvp (#119926)	2024-03-22 20:25:47 +00:00
init.h	[12/N] Apply clang-tidy and fix warnings in headers of torch/csrc (#116486)	2024-01-10 08:48:14 +00:00