reverting memory estimator for paganin to initial state and add odd s…

…ize for the test
DiamondLightSource · Sep 9, 2024 · 12949fd · 12949fd
1 parent 8dfcff4
commit 12949fd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/httomo/methods_database/packages/external/httomolibgpu/supporting_funcs/prep/phase.py b/httomo/methods_database/packages/external/httomolibgpu/supporting_funcs/prep/phase.py
@@ -101,7 +101,7 @@ def _calc_memory_bytes_paganin_filter_tomopy(
         + out_slice_size
         + 2 * complex_slice
         + 0.5 * fftplan_slice
-        + res_slice
+        + 2 * res_slice
     )
     subtract_bytes = int(filter_size + grid_size)
 

diff --git a/tests/test_backends/test_httomolibgpu.py b/tests/test_backends/test_httomolibgpu.py
@@ -155,8 +155,8 @@ def test_remove_outlier_memoryhook(flats, ensure_clean_memory, dtype, slices):
 
 @pytest.mark.cupy
 @pytest.mark.parametrize("slices", [2, 4, 8])
-@pytest.mark.parametrize("dim_x", [1024, 2048])
-@pytest.mark.parametrize("dim_y", [1024, 2048])
+@pytest.mark.parametrize("dim_x", [81, 1024, 2048])
+@pytest.mark.parametrize("dim_y", [345, 1024, 2048])
 def test_paganin_filter_tomopy_memoryhook(slices, dim_x, dim_y, ensure_clean_memory):
     data = cp.random.random_sample((slices, dim_x, dim_y), dtype=np.float32)
     hook = MaxMemoryHook()