Revert D48533397: disable recording memory snapshots after dumping

Differential Revision:
D48533397

Original commit changeset: cbf260823172

Original Phabricator Diff: D48533397

fbshipit-source-id: 6ef669973058fc9dc20f3b2839f4d931c3a58c3d

d2go/utils/gpu_memory_profiler.py

@@ -65,8 +65,6 @@ def log_memory_snapshot(output_dir: str, file_prefix: str = "") -> None:
         logger.info(f"Logging memory snapshot to {save_dir}")
         snapshot = torch.cuda.memory._snapshot()
         dump_snapshot(save_dir, snapshot)
-        # disable recording memory history
-        torch.cuda.memory._record_memory_history(enabled=None)
     except Exception as e:
         logger.error(f"Failed to log memory snapshot to {save_dir}: {e}")