Skip to content

Commit 637e4eb

Browse files
authored
Fix 2 cuda.bindings.nvml tests (#1727)
1 parent b65de83 commit 637e4eb

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

cuda_bindings/tests/nvml/test_device.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def test_get_nv_link_supported_bw_modes(all_devices):
7272
for device in all_devices:
7373
with unsupported_before(device, None):
7474
modes = nvml.device_get_nvlink_supported_bw_modes(device)
75-
assert isinstance(modes, nvml.NvLinkSupportedBWModes_v1)
75+
assert isinstance(modes, nvml.NvlinkSupportedBWModes_v1)
7676
# #define NVML_NVLINK_TOTAL_SUPPORTED_BW_MODES 23
7777
assert len(modes.bw_modes) <= 23
7878
assert not hasattr(modes, "total_bw_modes")
@@ -132,16 +132,6 @@ def test_read_write_prm(all_devices):
132132
assert isinstance(result[1], bytes)
133133

134134

135-
def test_nvlink_low_power_threshold(all_devices):
136-
for device in all_devices:
137-
# Docs say supported on HOPPER or newer
138-
with unsupported_before(device, None):
139-
try:
140-
nvml.device_set_nvlink_device_low_power_threshold(device, 0)
141-
except nvml.NoPermissionError:
142-
pytest.skip("No permission to set NVLink low power threshold")
143-
144-
145135
def test_get_power_management_limit(all_devices):
146136
for device in all_devices:
147137
# Docs say supported on KEPLER or later

0 commit comments

Comments
 (0)