enable GPU tests

GridTools · Nov 14, 2023 · b467a02 · b467a02
1 parent f1e1e02
commit b467a02
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/tests/next_tests/exclusion_matrices.py b/tests/next_tests/exclusion_matrices.py
@@ -50,6 +50,7 @@ class ProgramBackendId(_PythonObjectIdMixin, str, enum.Enum):
     GTFN_CPU_WITH_TEMPORARIES = (
         "gt4py.next.program_processors.runners.gtfn.run_gtfn_with_temporaries"
     )
+    GTFN_GPU = "gt4py.next.program_processors.runners.gtfn.run_gtfn_gpu"
     ROUNDTRIP = "gt4py.next.program_processors.runners.roundtrip.backend"
     DOUBLE_ROUNDTRIP = "gt4py.next.program_processors.runners.double_roundtrip.backend"
 

diff --git a/tests/next_tests/integration_tests/feature_tests/ffront_tests/ffront_test_utils.py b/tests/next_tests/integration_tests/feature_tests/ffront_tests/ffront_test_utils.py
@@ -33,6 +33,11 @@
     else:
         raise e
 
+try:
+    import cupy as cp
+except ImportError:
+    cp = None
+
 import next_tests
 import next_tests.exclusion_matrices as definitions
 
@@ -45,6 +50,9 @@ def no_backend(program: itir.FencilDefinition, *args: Any, **kwargs: Any) -> Non
 OPTIONAL_PROCESSORS = []
 if dace_iterator:
     OPTIONAL_PROCESSORS.append(definitions.OptionalProgramBackendId.DACE_CPU)
+GPU_PROCESSORS = []
+if cp:
+    GPU_PROCESSORS.append(definitions.ProgramBackendId.GTFN_GPU)
 
 
 @pytest.fixture(
@@ -54,7 +62,8 @@ def no_backend(program: itir.FencilDefinition, *args: Any, **kwargs: Any) -> Non
         definitions.ProgramBackendId.GTFN_CPU_IMPERATIVE,
         definitions.ProgramBackendId.GTFN_CPU_WITH_TEMPORARIES,
     ]
-    + OPTIONAL_PROCESSORS,
+    + OPTIONAL_PROCESSORS
+    + GPU_PROCESSORS,
     ids=lambda p: p.short_id() if p is not None else "None",
 )
 def fieldview_backend(request):