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 index b71163e694..1c42a9db47 100644 --- 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 @@ -50,9 +50,6 @@ 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( @@ -61,9 +58,9 @@ def no_backend(program: itir.FencilDefinition, *args: Any, **kwargs: Any) -> Non definitions.ProgramBackendId.GTFN_CPU, definitions.ProgramBackendId.GTFN_CPU_IMPERATIVE, definitions.ProgramBackendId.GTFN_CPU_WITH_TEMPORARIES, + pytest.param(definitions.ProgramBackendId.GTFN_GPU, marks=pytest.mark.requires_gpu), ] - + OPTIONAL_PROCESSORS - + GPU_PROCESSORS, + + OPTIONAL_PROCESSORS, ids=lambda p: p.short_id() if p is not None else "None", ) def fieldview_backend(request): diff --git a/tests/next_tests/unit_tests/conftest.py b/tests/next_tests/unit_tests/conftest.py index b43eeb3f91..ce0fded59c 100644 --- a/tests/next_tests/unit_tests/conftest.py +++ b/tests/next_tests/unit_tests/conftest.py @@ -60,6 +60,7 @@ def lift_mode(request): (definitions.ProgramBackendId.GTFN_CPU, True), (definitions.ProgramBackendId.GTFN_CPU_IMPERATIVE, True), (definitions.ProgramBackendId.GTFN_CPU_WITH_TEMPORARIES, True), + pytest.param((definitions.ProgramBackendId.GTFN_GPU, True), marks=pytest.mark.requires_gpu), (definitions.ProgramFormatterId.LISP_FORMATTER, False), (definitions.ProgramFormatterId.ITIR_PRETTY_PRINTER, False), (definitions.ProgramFormatterId.ITIR_TYPE_CHECKER, False),