aparler/c-llm
1
0
Fork 0
mirror of https://github.com/karpathy/llm.c.git synced 2026-07-28 20:35:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

added missing free

Browse source
This commit is contained in:
Erik Schultheis 2024-06-25 00:31:08 +03:00
parent 8b36c2f60b
commit 39f976f013
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
train_gpt2.cu
View file

@ -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);