Skip to content

Commit 432a36e

Browse files
committed
Add calc_peak_gpu_mem to paganin_filter_savu_legacy
1 parent 07df76c commit 432a36e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

httomolibgpu/prep/phase.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,6 +298,7 @@ def paganin_filter_savu_legacy(
298298
distance: float = 1.0,
299299
energy: float = 53.0,
300300
ratio_delta_beta: float = 250,
301+
calc_peak_gpu_mem: bool = False,
301302
) -> cp.ndarray:
302303
"""
303304
Perform single-material phase retrieval from flats/darks corrected tomographic measurements. For more detailed information, see :ref:`phase_contrast_module`.
@@ -323,4 +324,4 @@ def paganin_filter_savu_legacy(
323324
The 3D array of Paganin phase-filtered projection images.
324325
"""
325326

326-
return paganin_filter(tomo, pixel_size, distance, energy, ratio_delta_beta / 4)
327+
return paganin_filter(tomo, pixel_size, distance, energy, ratio_delta_beta / 4, calc_peak_gpu_mem=calc_peak_gpu_mem)

0 commit comments

Comments
 (0)