Skip to content

Commit c3b5c91

Browse files
skip test if no pycuda or no cuda device
1 parent 283d8ef commit c3b5c91

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/test_energy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,11 @@
11
import os
2-
2+
from .context import skip_if_no_pycuda
33
from kernel_tuner.energy import energy
44

55

66
cache_filename = os.path.dirname(os.path.realpath(__file__)) + "/synthetic_fp32_cache_NVIDIA_RTX_A4000.json"
77

8+
@skip_if_no_pycuda
89
def test_create_power_frequency_model():
910

1011
ridge_frequency, freqs, nvml_power, fitted_params, scaling = energy.create_power_frequency_model(cache=cache_filename)

0 commit comments

Comments
 (0)