Improve memory usage in test_bert_batch_dram.py

tenstorrent · Dec 11, 2024 · 2abd3fe · 2abd3fe
1 parent 8b82811
commit 2abd3fe
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/models/demos/metal_BERT_large_11/tests/test_bert_batch_dram.py b/models/demos/metal_BERT_large_11/tests/test_bert_batch_dram.py
@@ -221,6 +221,13 @@ def run_bert_question_and_answering_inference(
     profiler.end("processing_output_to_string")
 
     del tt_out
+    del tt_embedding
+    del tt_attention_mask
+    del tt_embedding_inputs
+    del bert_input
+    del pytorch_out
+    if "single_inputs" in locals():
+        del single_inputs
 
     profiler.print()