fwiw, i checked whether this tutorial was working on the PyTorch Nightly (by running this script directly: https://
raw.githubusercontent.com/pytorch/tutori
als/main/recipes_source/recipes/profiler_recipe.py
… )
and on the Nightly, this seem to have been fixed.
So it was either fixed in 2.1/2.2; or it'll be fixed in the next release.

→ View original post on X — @soumithchintala