diff options
| author | Gemma Fardell <47746591+gfardell@users.noreply.github.com> | 2019-12-17 13:18:46 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-12-17 13:18:46 +0000 |
| commit | 190a5a7679c506e9fc78969ec0f1df4f730aa6f7 (patch) | |
| tree | 67695949a4d904b89b70c4257c30471622ea9590 | |
| parent | cdca2ba12a9dd3df67a00d0d3f40497aacadbe2b (diff) | |
| parent | 76374867549ee85e89478ce894875ca27217f836 (diff) | |
| download | astra-wrapper-190a5a7679c506e9fc78969ec0f1df4f730aa6f7.tar.gz astra-wrapper-190a5a7679c506e9fc78969ec0f1df4f730aa6f7.tar.bz2 astra-wrapper-190a5a7679c506e9fc78969ec0f1df4f730aa6f7.tar.xz astra-wrapper-190a5a7679c506e9fc78969ec0f1df4f730aa6f7.zip | |
Merge pull request #48 from vais-ral/deallocate_cuda_memory-1
deallocates memory on GPU
| -rw-r--r-- | Wrappers/Python/ccpi/astra/processors/FBP.py | 5 |
1 files changed, 3 insertions, 2 deletions
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 |