torch_memory_saver
torch_memory_saver_hook_mode_preload
torch_memory_saver_hook_mode_torch