WIP

neon60 · neon60 · commit a59c6c1aa3ce · 2025-01-13T08:56:41.000+01:00
diff --git a/httomolibgpu/recon/rotation.py b/httomolibgpu/recon/rotation.py
@@ -344,7 +344,8 @@ def _calculate_chunks(
 
     available_memory -= shift_size
     freq_domain_size = (
-        shift_size  # it needs only half (RFFT), but complex64, so it's the same
+        # shift_size  # it needs only half (RFFT), but complex64, so it's the same
+        shift_size * 2  # it needs full (FFT), with complex64, so it's double
     )
     fft_plan_size = freq_domain_size
     size_per_shift = 2 * (fft_plan_size + freq_domain_size + shift_size)

Original file line number	Diff line number	Diff line change
`@@ -344,7 +344,8 @@ def _calculate_chunks(`
`344`	`344`
`345`	`345`	`available_memory -= shift_size`
`346`	`346`	`freq_domain_size = (`
`347`		`- shift_size # it needs only half (RFFT), but complex64, so it's the same`
	`347`	`+ # shift_size # it needs only half (RFFT), but complex64, so it's the same`
	`348`	`+ shift_size * 2 # it needs full (FFT), with complex64, so it's double`
`348`	`349`	`)`
`349`	`350`	`fft_plan_size = freq_domain_size`
`350`	`351`	`size_per_shift = 2 * (fft_plan_size + freq_domain_size + shift_size)`