Skip to content

Commit 34ad050

Browse files
committed
Do not assume fft plan cache is cleared
1 parent 614b773 commit 34ad050

File tree

1 file changed

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

1 file changed

+6
-6
lines changed

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

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -248,12 +248,12 @@ def _calc_memory_bytes_LPRec3d_tomobar(
248248
tot_memory_bytes = int( max(
249249
in_slice_size + padded_in_slice_size + recon_output_size + tmp_p_input_slice + padded_tmp_p_input_slice + rfft_plan_slice_size + rfft_result_size + filtered_rfft_result_size
250250
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + tmp_p_input_slice + rfft_result_size
251-
, in_slice_size + padded_in_slice_size + recon_output_size + tmp_p_input_slice + irfft_plan_slice_size + irfft_scratch_memory_size + irfft_result_size
252-
, in_slice_size + padded_in_slice_size + recon_output_size + tmp_p_input_slice + irfft_plan_slice_size + datac_size + conversion_to_complex_size
253-
, in_slice_size + padded_in_slice_size + recon_output_size + irfft_plan_slice_size + datac_size + fde_size + shifted_datac_size + fft_plan_slice_size + fft_result_size + backshifted_datac_size + scaled_backshifted_datac_size
254-
, in_slice_size + padded_in_slice_size + recon_output_size + irfft_plan_slice_size + fde_size + fft_plan_slice_size + ifft2_plan_slice_size
255-
, in_slice_size + padded_in_slice_size + recon_output_size + irfft_plan_slice_size + fft_plan_slice_size + ifft2_plan_slice_size + fde_view_size + concatenate_size
256-
, in_slice_size + padded_in_slice_size + recon_output_size + irfft_plan_slice_size + fft_plan_slice_size + ifft2_plan_slice_size + after_recon_swapaxis_slice
251+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + tmp_p_input_slice + irfft_plan_slice_size + irfft_scratch_memory_size + irfft_result_size
252+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + tmp_p_input_slice + irfft_plan_slice_size + datac_size + conversion_to_complex_size
253+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + irfft_plan_slice_size + datac_size + fde_size + shifted_datac_size + fft_plan_slice_size + fft_result_size + backshifted_datac_size + scaled_backshifted_datac_size
254+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + irfft_plan_slice_size + fde_size + fft_plan_slice_size + ifft2_plan_slice_size
255+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + irfft_plan_slice_size + fft_plan_slice_size + ifft2_plan_slice_size + fde_view_size + concatenate_size
256+
, in_slice_size + padded_in_slice_size + recon_output_size + rfft_plan_slice_size + irfft_plan_slice_size + fft_plan_slice_size + ifft2_plan_slice_size + after_recon_swapaxis_slice
257257
)
258258
)
259259

0 commit comments

Comments
 (0)