[Profiler] Fix memory profiler merge issue (#60037)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60037
The memory profiler was broken due to a mis-merge during rebase. Add lost line back.
Reviewed By: ezyang
Differential Revision: D29143469
fbshipit-source-id: c3bf0088ca12e7535eeddbede24e28201eccd5f4
