Merge pull request #203 from KernelTuner/skip_nvml_tests

Skip nvml tests

Author: benvanwerkhoven

test/context.py

@@ -11,6 +11,12 @@
 except Exception:
     pycuda_present = False
 
+try:
+    import pynvml
+    pynvml_present = True
+except ImportError:
+    pynvml_present = False
+
 try:
     import pyopencl
     opencl_present = True
@@ -39,6 +45,7 @@
     cuda_present = False
 
 skip_if_no_pycuda = pytest.mark.skipif(not pycuda_present, reason="PyCuda not installed or no CUDA device detected")
+skip_if_no_pynvml = pytest.mark.skipif(not pynvml_present, reason="NVML not installed")
 skip_if_no_cupy = pytest.mark.skipif(not cupy_present, reason="CuPy not installed or no CUDA device detected")
 skip_if_no_cuda = pytest.mark.skipif(not cuda_present, reason="NVIDIA CUDA not installed")
 skip_if_no_opencl = pytest.mark.skipif(not opencl_present, reason="PyOpenCL not installed or no OpenCL device detected")

test/test_energy.py

@@ -1,11 +1,12 @@
 import os
-from .context import skip_if_no_pycuda
+from .context import skip_if_no_pycuda, skip_if_no_pynvml
 from kernel_tuner.energy import energy
 
 
 cache_filename = os.path.dirname(os.path.realpath(__file__)) + "/synthetic_fp32_cache_NVIDIA_RTX_A4000.json"
 
 @skip_if_no_pycuda
+@skip_if_no_pynvml
 def test_create_power_frequency_model():
 
     ridge_frequency, freqs, nvml_power, fitted_params, scaling = energy.create_power_frequency_model(cache=cache_filename, simulation_mode=True)

test/test_observers.py

@@ -5,11 +5,12 @@
 from kernel_tuner.observers.nvml import NVMLObserver
 from kernel_tuner.observers.observer import BenchmarkObserver
 
-from .context import skip_if_no_pycuda
+from .context import skip_if_no_pycuda, skip_if_no_pynvml
 from .test_runners import env
 
 
 @skip_if_no_pycuda
+@skip_if_no_pynvml
 def test_nvml_observer(env):
     nvmlobserver = NVMLObserver(["nvml_energy", "temperature"])
     env[-1]["block_size_x"] = [128]