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 [1/n] Support Dynamic Memory Budget in Auto AC (#143539) 2024-12-21 07:38:52 +00:00
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 [functorch] clean up asserts in test_dims.py (#144276) 2025-01-08 01:21:40 +00:00
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 Remove det_singular OpInfo (#145533) 2025-01-25 00:58:03 +00:00
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 PEP585 update - test (#145176) 2025-01-22 04:48:28 +00:00
test_vmap.py Fix unbind_copy and add its decomposition (#134319) 2025-01-17 18:21:22 +00:00
test_vmap_registrations.py
xfail_suggester.py