Skip to content

Commit 2717987

Browse files
authored
Comment out memory snapshot dump in profiler.py
Commented out the memory snapshot dump line in profiler.py.
1 parent 85f79ff commit 2717987

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmarks/microbenchmarks/profiler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def generate_memory_profile(model, input_data, profile_file_path):
124124
torch.cuda.synchronize()
125125

126126
# Take memory snapshot after inference and save to temporary pickle file
127-
torch.cuda.memory._dump_snapshot(profile_file_path)
127+
# torch.cuda.memory._dump_snapshot(profile_file_path)
128128

129129
if _validate_pickle_file(profile_file_path):
130130
print(f"Saved memory profile to {profile_file_path}")

0 commit comments

Comments
 (0)