Edward Z. Yang 7aa6bd7fa0 Refactor all top level usages of record_shapeenv_event to ShapeEnv class (#123735) ... This ensures that first argument to record_shapeenv_event is a ShapeEnv so we can appropriately short circuit when recording is not in progress. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/123735 Approved by: https://github.com/ysiraichi, https://github.com/zou3519, https://github.com/albanD		2024-04-27 20:36:40 +00:00
..
__init__.py	Add structured trace logs (#120289)	2024-02-28 01:01:41 +00:00
_internal.py	[compiled autograd] introduce verbose logs, add autograd node info to graph (#124954)	2024-04-27 01:10:37 +00:00
_registrations.py	Refactor all top level usages of record_shapeenv_event to ShapeEnv class (#123735)	2024-04-27 20:36:40 +00:00
structured.py	Add structured trace logs (#120289)	2024-02-28 01:01:41 +00:00