fixed one more memoy leak

cjknight · Aug 9, 2024 · c36888d · c36888d
1 parent 272db99
commit c36888d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gpu/src/device_cuda.cpp b/gpu/src/device_cuda.cpp
@@ -1793,6 +1793,7 @@ cublasDgemm(dd->handle, CUBLAS_OP_T, CUBLAS_OP_N,
   pm->dev_free(d_bPvu);
   pm->dev_free(d_vuwm);
   pm->dev_free(d_vuwM);
+  free(h_vuwM);
 profile_stop();
 #ifdef _SIMPLE_TIMER
   double t1 = omp_get_wtime();