pytorch/test/functorch
rzou 15b1ac3e86 Add torch.func.debug_unwrap (#146528)
Use it to unwrap any functorch-wrapped tensor. I don't recommend using
the output in a program since it breaks the semantics of the transforms,
but it seems useful for debugging.

I will note that some people have wanted to get intermediate values out
of an e.g. grad transform, so this might be a way to do that...

Test Plan:
- tests
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146528
Approved by: https://github.com/Chillee
2025-02-06 18:48:09 +00:00
..
attn_ft.py
attn_positional.py
common_utils.py
discover_coverage.py Apply ruff fixes to tests (#146140) 2025-02-04 05:41:01 +00:00
functorch_additional_op_db.py
test_ac.py
test_ac_knapsack.py
test_aotdispatch.py Remove FFT from stride incorrect ops (#145080) 2025-01-27 04:26:04 +00:00
test_control_flow.py Add non-strict export while_loop test back (#146195) 2025-02-03 19:28:22 +00:00
test_dims.py
test_eager_transforms.py Add torch.func.debug_unwrap (#146528) 2025-02-06 18:48:09 +00:00
test_logging.py
test_memory_efficient_fusion.py
test_minifier.py
test_ops.py
test_parsing.py [BE][Ez]: Enable ruff rule E731. use def instead of anonymous lambda (#146410) 2025-02-05 01:44:41 +00:00
test_rearrange.py
test_vmap.py
test_vmap_registrations.py
xfail_suggester.py