Skip to content

Commit ea18454

Browse files
committed
a small increase of the memory threshold for rescale_to_int
1 parent 276025a commit ea18454

File tree

1 file changed

+5
-2
lines changed
  • httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/misc

1 file changed

+5
-2
lines changed

httomo_backends/methods_database/packages/backends/httomolibgpu/supporting_funcs/misc/rescale.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,11 @@ def _calc_memory_bytes_rescale_to_int(
1818
itemsize = 2
1919
else:
2020
itemsize = 4
21-
safety = 128
21+
safety_multiplier = 1.1
2222
return (
23-
int(np.prod(non_slice_dims_shape)) * (dtype.itemsize + itemsize) + safety,
23+
int(
24+
safety_multiplier
25+
* ((np.prod(non_slice_dims_shape)) * (dtype.itemsize + itemsize))
26+
),
2427
0,
2528
)

0 commit comments

Comments
 (0)