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

Updated Dispatcher Structured Kernels Lab (markdown)

Brian Hirsh
2022-04-19 09:17:25 -04:00
parent 7e26a8aead
commit a2fa600791
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Dispatcher---Structured-Kernels-Lab.md

@ -86,7 +86,9 @@ You can re-run the gdb the same way that we did before and see that a lot less i
Another feature that’s provided directly inside of the dispatcher is the profiler, which profiles the set of aten ops that ran in your model as well as how much time was spent in them.
Run the following snippet before/after your change, and compare:
Run the following snippet before/after your change, and compare
(Also, make sure that you built with `REL_WITH_DEB_INFO=1`, so you're using a release builds. Otherwise your times will be unnecessarily slow.)
>>> import torch
>>> with torch.profiler.profile() as prof: