Fix memory leak in CudaANISymmetryFunctions

openmm · Oct 6, 2020 · 377214d · 377214d
1 parent d78dc23
commit 377214d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ani/CudaANISymmetryFunctions.cu b/ani/CudaANISymmetryFunctions.cu
@@ -76,6 +76,8 @@ CudaANISymmetryFunctions::~CudaANISymmetryFunctions() {
         cudaFree(positions);
     if (neighbors != 0)
         cudaFree(neighbors);
+    if (neighborCount != 0)
+        cudaFree(neighborCount);
     if (periodicBoxVectors != 0)
         cudaFree(periodicBoxVectors);
     if (angularIndex != 0)