Fix memory leak

grandinetech · Jun 5, 2024 · 1c65661 · 1c65661
1 parent e1128ec
commit 1c65661
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/arkworks3-sppark-wlc/cuda/pippenger_inf.cu b/arkworks3-sppark-wlc/cuda/pippenger_inf.cu
@@ -549,7 +549,9 @@ RustError mult_pippenger_faster_inf(RustContext<bucket_t, affine_t, scalar_t> *c
 // Allocate device storage, transfer bases
 extern "C" 
 void mult_pippenger_faster_free(RustContext<bucket_t, affine_t, scalar_t> *context) {
-    free(context->context);
+    cudaFreeHost(context->context->h_scalars);
+    cudaFreeHost(context->context->h_scalar_map);
+    delete context->context;
 }
 
 #endif  //  __CUDA_ARCH__
diff --git a/arkworks3-sppark-wlc/src/lib.rs b/arkworks3-sppark-wlc/src/lib.rs
@@ -45,8 +45,6 @@ extern "C" {
 }
 
 pub fn multi_scalar_mult_init<G: AffineCurve>(points: &[G]) -> MultiScalarMultContext {
-    println!("INIT!");
-
     let mut ret = MultiScalarMultContext {
         context: std::ptr::null_mut(),
     };
@@ -67,8 +65,6 @@ pub fn multi_scalar_mult_init<G: AffineCurve>(points: &[G]) -> MultiScalarMultCo
 }
 
 pub fn multi_scalar_mult_free(context: &mut MultiScalarMultContext) {
-    println!("FREE!");
-
     unsafe {
         mult_pippenger_faster_free(context);
     }