Skip to content

Commit 3eebbcf

Browse files
only test if pycuda is installed
1 parent b2ca886 commit 3eebbcf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_energy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

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

8-
8+
@skip_if_no_pycuda
99
def test_create_power_frequency_model():
1010

1111
ridge_frequency, freqs, nvml_power, fitted_params, scaling = energy.create_power_frequency_model(cache=cache_filename, simulation_mode=True)

0 commit comments

Comments
 (0)