frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] Added debug commands

Browse Source
This commit is contained in:
Horace He
2021-04-29 14:20:26 -07:00
committed by Jon Janzen
parent 033e6e30af
commit 03550ccd61
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
functorch/README.md
View File

@ -156,6 +156,10 @@ batched jacobians:
>>> hessian = jacrev(jacrev(f))(x)
```
## Debugging
`functorch._C.dump_tensor`: Dumps dispatch keys on stack
`torch._C._debug_only_display_vmap_fallback_warnings(True)`: Shows vmap fallbacks to loop/stack
## Future Plans
In the end state, we'd like to upstream this into PyTorch once we iron out the