Skip to content

Commit e5e30e8

Browse files
authored
test(nvml): skip unrepairable memory flag test on devices that do not support it (#1560)
1 parent 4408aef commit e5e30e8

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

cuda_bindings/tests/nvml/test_device.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,8 @@ def test_device_get_performance_modes(all_devices):
9494
@pytest.mark.skipif(cuda_version_less_than(13010), reason="Introduced in 13.1")
9595
def test_device_get_unrepairable_memory_flag(all_devices):
9696
for device in all_devices:
97-
status = nvml.device_get_unrepairable_memory_flag_v1(device)
97+
with unsupported_before(device, None):
98+
status = nvml.device_get_unrepairable_memory_flag_v1(device)
9899
assert isinstance(status, int)
99100

100101

0 commit comments

Comments
 (0)