File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/recon Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -219,7 +219,7 @@ def _calc_memory_bytes_LPRec3d_tomobar(
219219 if odd_horiz :
220220 output_dims = tuple (x + 1 for x in output_dims )
221221
222- in_slice_size = np . prod ( non_slice_dims_shape ) * dtype .itemsize
222+ in_slice_size = ( angles_tot * DetectorsLengthH ) * dtype .itemsize
223223 padded_in_slice_size = angles_tot * n * np .float32 ().itemsize
224224
225225 theta_size = angles_tot * np .float32 ().itemsize
@@ -352,9 +352,7 @@ def add_to_memory_counters(amount, per_slice: bool):
352352 add_to_memory_counters (circular_mask_size , False )
353353 add_to_memory_counters (after_recon_swapaxis_slice , True )
354354
355- return (int (tot_memory_bytes * 1.05 ), fixed_amount + 250 * 1024 * 1024 )
356- # return (tot_memory_bytes, fixed_amount)
357-
355+ return (tot_memory_bytes * 1.05 , fixed_amount + 250 * 1024 * 1024 )
358356
359357def _calc_memory_bytes_SIRT3d_tomobar (
360358 non_slice_dims_shape : Tuple [int , int ],
You can’t perform that action at this time.
0 commit comments