adding pytest marker

Author: Olender

tests/firedrake/conftest.py

@@ -7,7 +7,7 @@
 os.environ["FIREDRAKE_DISABLE_OPTIONS_LEFT"] = "1"
 
 import pytest
-from firedrake.petsc import PETSc, get_external_packages
+from firedrake.petsc import PETSc, get_external_packages, get_petsc_variables
 
 
 def _skip_test_dependency(dependency):
@@ -145,11 +145,21 @@ def pytest_configure(config):
         "markers",
         "skipnetgen: mark as skipped if netgen and ngsPETSc is not installed"
     )
+    config.addinivalue_line(
+        "markers",
+        "skipcuda: mark as skipped if CUDA is not available"
+    )
 
 
 def pytest_collection_modifyitems(session, config, items):
     from firedrake.utils import complex_mode, SLATE_SUPPORTS_COMPLEX
 
+    try:
+        get_petsc_variables()["CUDA_VERSION"]
+        cuda_unavailable = False
+    except:
+        cuda_unavailable = True
+
     for item in items:
         if complex_mode:
             if item.get_closest_marker("skipcomplex") is not None:
@@ -160,6 +170,10 @@ def pytest_collection_modifyitems(session, config, items):
             if item.get_closest_marker("skipreal") is not None:
                 item.add_marker(pytest.mark.skip(reason="Test makes no sense unless in complex mode"))
 
+        if cuda_unavailable:
+            if item.get_closest_marker("skipcuda") is not None:
+                item.add_marker(pytest.mark.skip(reason="CUDA not available"))
+
         for dep, marker, reason in dependency_skip_markers_and_reasons:
             if _skip_test_dependency(dep) and item.get_closest_marker(marker) is not None:
                 item.add_marker(pytest.mark.skip(reason))

tests/firedrake/cuda/test_poisson_offloading_pc.py

@@ -3,6 +3,7 @@
 import pytest
 
 # TODO: add marker for cuda pytests and something to check if cuda memory was really used
+@pytest.mark.skipcuda
 @pytest.mark.parametrize("ksp_type, pc_type", [("cg", "sor"), ("cg", "gamg"), ("preonly", "lu")])
 def test_poisson_on_cuda(ksp_type, pc_type):
 

tests/firedrake/cuda/test_wave_equation.py

@@ -6,7 +6,7 @@
 import pytest
 
 
-# TODO: add marker for cuda pytestss
+@pytest.mark.skipcuda
 def test_kmv_wave_propagation_cuda():
     nested_parameters = {
         "ksp_type": "preonly",