diff --git a/cuda_core/tests/system/test_system_device.py b/cuda_core/tests/system/test_system_device.py index 8bd272c539..45ec6570fa 100644 --- a/cuda_core/tests/system/test_system_device.py +++ b/cuda_core/tests/system/test_system_device.py @@ -770,16 +770,19 @@ def test_nvlink(): nvlink_info = device.get_nvlink(link) assert isinstance(nvlink_info, _device.NvlinkInfo) + with unsupported_before(device, None): + state = nvlink_info.state + assert isinstance(state, bool) + + if not state: + continue + with unsupported_before(device, None): version = nvlink_info.version assert isinstance(version, tuple) assert len(version) == 2 assert all(isinstance(i, int) for i in version) - with unsupported_before(device, None): - state = nvlink_info.state - assert isinstance(state, bool) - def test_utilization(): for device in system.Device.get_all_devices():