OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
52,104 Commits 8,304 Branches 150 Tags 1.5 GiB
3a3e2002d8
Commit Graph

2 Commits

Author SHA1 Message Date
Mateusz Sypniewski
67d6f7160c Add synchronize hooks (#84427) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84427
Approved by: https://github.com/ngimel, https://github.com/lw
 2022-09-09 13:56:59 +00:00
Mateusz Sypniewski
916def84d4 CUDA trace Python hooks (#82824) 
### Description
This adds Python hooks into PyTorch that allow the user to register their own callbacks for events such as tensor allocation, stream allocation, event record / wait etc.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82824
Approved by: https://github.com/lw, https://github.com/ezyang, https://github.com/malfet
 2022-08-11 10:21:40 +00:00