From b9f3c848bbd46d0cdb6d583fb3d8ec46696fe5da Mon Sep 17 00:00:00 2001
From: chaliebudd <charles.budd@kcl.ac.uk>
Date: Wed, 19 May 2021 16:06:21 +0100
Subject: [PATCH] cleaning allocated memory

Signed-off-by: chaliebudd <charles.budd@kcl.ac.uk>
---
 monai/csrc/filtering/permutohedral/permutohedral_cuda.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/monai/csrc/filtering/permutohedral/permutohedral_cuda.cu b/monai/csrc/filtering/permutohedral/permutohedral_cuda.cu
index b87a88a84f..0977b3f141 100644
--- a/monai/csrc/filtering/permutohedral/permutohedral_cuda.cu
+++ b/monai/csrc/filtering/permutohedral/permutohedral_cuda.cu
@@ -529,6 +529,8 @@ void PermutohedralCuda(scalar_t* values, scalar_t* positions, int elementCount,
 
   destroyHashTable<scalar_t>();
   cudaFree(table_values);
+  cudaFree(scaleFactor);
+  cudaFree(matrix);
 }
 
 #define DECLARATION(dc, fc)                                                                                         \