fix(multilinear-extensions): address concerns of double free on copy

mratsim · Jul 6, 2024 · acbfc48 · acbfc48
1 parent a401fc8
commit acbfc48
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/constantine/boolean_hypercube/multilinear_extensions.nim b/constantine/boolean_hypercube/multilinear_extensions.nim
@@ -40,6 +40,9 @@ type
     num_vars*: int
     base_poly_evals*: ptr UncheckedArray[F]
 
+proc `=wasMoved`*[F](mle: var MultilinearExtension[F]) =
+  mle.base_poly_evals = nil
+
 proc `=destroy`*[F](mle: var MultilinearExtension[F]) =
   if not mle.base_poly_evals.isNil:
     freeHeapAligned(mle.base_poly_evals)