remove cache - might make it slower but should run

C2SM · Nov 27, 2024 · 4bc8ad1 · 4bc8ad1
1 parent 512e3cf
commit 4bc8ad1
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 9 deletions.
diff --git a/model/atmosphere/diffusion/tests/diffusion_tests/test_diffusion.py b/model/atmosphere/diffusion/tests/diffusion_tests/test_diffusion.py
@@ -61,9 +61,7 @@ def _construct_minimal_decomposition_info(grid: icon.IconGrid):
         return decomposition_info
 
     if not grid_functionality[experiment].get(name):
-        on_gpu = helpers.is_gpu(backend)
-        gm = grid_utils.get_icon_grid_from_gridfile(experiment, on_gpu)
-        gm()
+        gm = grid_utils.get_icon_grid_from_gridfile(experiment, backend)
         grid = gm.grid
         decomposition_info = _construct_minimal_decomposition_info(grid)
         geometry_ = geometry.GridGeometry(

diff --git a/model/common/src/icon4py/model/common/test_utils/grid_utils.py b/model/common/src/icon4py/model/common/test_utils/grid_utils.py
@@ -6,7 +6,6 @@
 # Please, refer to the LICENSE file in the root directory.
 # SPDX-License-Identifier: BSD-3-Clause
 
-import functools
 
 import gt4py.next.backend as gtx_backend
 import pytest
@@ -25,7 +24,6 @@
 MCH_CH_R04B09_LEVELS = 65
 
 
-@functools.cache
 def get_icon_grid_from_gridfile(
     experiment: str, backend: gtx_backend.Backend = False
 ) -> gm.GridManager:

diff --git a/model/common/src/icon4py/model/common/test_utils/helpers.py b/model/common/src/icon4py/model/common/test_utils/helpers.py
@@ -44,10 +44,6 @@ def is_embedded(backend) -> bool:
     return backend is None
 
 
-def is_gpu(backend) -> bool:
-    return "gpu" in backend.name if backend else False
-
-
 def is_roundtrip(backend) -> bool:
     return backend.name == "roundtrip" if backend else False