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

mratsim · Jul 6, 2024 · ec88a5c · ec88a5c
1 parent a401fc8
commit ec88a5c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/constantine/boolean_hypercube/multilinear_extensions.nim b/constantine/boolean_hypercube/multilinear_extensions.nim
@@ -40,6 +40,11 @@ type
     num_vars*: int
     base_poly_evals*: ptr UncheckedArray[F]
 
+proc `=copy`*[F](dst: var MultilinearExtension[F], src: MultilinearExtension[F]) {.error: "A multilinear extension cannot be copied".}
+
+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)