From 76374867549ee85e89478ce894875ca27217f836 Mon Sep 17 00:00:00 2001 From: Edoardo Pasca Date: Tue, 17 Dec 2019 13:17:07 +0000 Subject: deallocates memory on GPU closes #47 --- Wrappers/Python/ccpi/astra/processors/FBP.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'Wrappers/Python/ccpi') diff --git a/Wrappers/Python/ccpi/astra/processors/FBP.py b/Wrappers/Python/ccpi/astra/processors/FBP.py index adc4119..adf27eb 100644 --- a/Wrappers/Python/ccpi/astra/processors/FBP.py +++ b/Wrappers/Python/ccpi/astra/processors/FBP.py @@ -242,9 +242,10 @@ class FBP(DataProcessor): # since we run gpu we need to rescale and flip it IM.array[i] = np.flip(astra.data2d.get(rec_id) * self.volume_geometry.voxel_size_x,0) - astra.algorithm.delete(alg_id) + astra.algorithm.delete(alg_id) + astra.data2d.delete(sinogram_id) + astra.data2d.delete(rec_id) - astra.data2d.delete(sinogram_id) return IM -- cgit v1.2.3