Skip to content

Commit 2700905

Browse files
committed
Improve for min mem options at LPRec
1 parent 9258aac commit 2700905

File tree

1 file changed

+3
-3
lines changed
  • httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/recon

1 file changed

+3
-3
lines changed

httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/recon/algorithm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -313,8 +313,8 @@ def add_to_memory_counters(amount, per_slice: bool):
313313

314314
add_to_memory_counters(tmp_p_input_slice, True)
315315
if min_mem_usage_filter:
316-
add_to_memory_counters(rfft_plan_slice_size * 2, False)
317-
add_to_memory_counters(irfft_plan_slice_size * 2, False)
316+
add_to_memory_counters(rfft_plan_slice_size / 4, False)
317+
add_to_memory_counters(irfft_plan_slice_size / 4, False)
318318
add_to_memory_counters(padded_tmp_p_input_slice, False)
319319

320320
add_to_memory_counters(rfft_result_size, False)
@@ -362,7 +362,7 @@ def add_to_memory_counters(amount, per_slice: bool):
362362
add_to_memory_counters(-fft_result_size, True)
363363

364364
if min_mem_usage_ifft2:
365-
add_to_memory_counters(ifft2_plan_slice_size * 2, False)
365+
add_to_memory_counters(ifft2_plan_slice_size, False)
366366
add_to_memory_counters(fde_size * 2, False)
367367
add_to_memory_counters(-fde_size * 2, False)
368368
else:

0 commit comments

Comments
 (0)