From 39f976f0134308b198594ecccb24a19a8b4e7da8 Mon Sep 17 00:00:00 2001
From: Erik Schultheis <erik.schultheis@aalto.fi>
Date: Tue, 25 Jun 2024 00:31:08 +0300
Subject: [PATCH] added missing free

---
 train_gpt2.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train_gpt2.cu b/train_gpt2.cu
index d4ed0f1..1412df4 100644
--- a/train_gpt2.cu
+++ b/train_gpt2.cu
@@ -1140,6 +1140,7 @@ void gpt2_free(GPT2 *model) {
     cudaFreeCheck(&model->acts_memory);
     cudaFreeCheck(&model->inputs);
     cudaFreeCheck(&model->targets);
+    cudaFreeCheck(&model->accumulated_mean_loss);
     cudaCheck(cudaFreeHost(model->cpu_losses));
     cudaCheck(cudaFreeHost(model->cpu_losses_fp32));
     free(model->workload_indices);