diff --git a/cuda_bindings/tests/nvml/test_nvlink.py b/cuda_bindings/tests/nvml/test_nvlink.py index 04bc8eaae4c..1ea9e25dfa7 100644 --- a/cuda_bindings/tests/nvml/test_nvlink.py +++ b/cuda_bindings/tests/nvml/test_nvlink.py @@ -27,8 +27,3 @@ def test_nvlink_get_link_count(all_devices): assert value.nvml_return == nvml.Return.SUCCESS or value.nvml_return == nvml.Return.ERROR_NOT_SUPPORTED, ( f"Unexpected return {value.nvml_return} for link count field query" ) - - # The feature_nvlink_supported detection is not robust, so we - # can't be more specific about how many links we should find. - if value.nvml_return == nvml.Return.SUCCESS: - assert value.value.ui_val[0] <= nvml.NVLINK_MAX_LINKS, f"Unexpected link count {value.value.ui_val[0]}" diff --git a/cuda_bindings/tests/nvml/test_pynvml.py b/cuda_bindings/tests/nvml/test_pynvml.py index 2d1029e9d9b..ad8eb4c63fa 100644 --- a/cuda_bindings/tests/nvml/test_pynvml.py +++ b/cuda_bindings/tests/nvml/test_pynvml.py @@ -265,27 +265,6 @@ def test_device_get_pcie_throughput(ngpus, handles): # Test pynvml.nvmlDeviceGetNvLinkRemotePciInfo -@pytest.mark.parametrize( - "cap_type", - [ - nvml.NvLinkCapability.NVLINK_CAP_P2P_SUPPORTED, # P2P over NVLink is supported - nvml.NvLinkCapability.NVLINK_CAP_SYSMEM_ACCESS, # Access to system memory is supported - nvml.NvLinkCapability.NVLINK_CAP_P2P_ATOMICS, # P2P atomics are supported - nvml.NvLinkCapability.NVLINK_CAP_SYSMEM_ATOMICS, # System memory atomics are supported - nvml.NvLinkCapability.NVLINK_CAP_SLI_BRIDGE, # SLI is supported over this link - nvml.NvLinkCapability.NVLINK_CAP_VALID, - ], -) # Link is supported on this device -def test_device_get_nvlink_capability(ngpus, handles, cap_type): - for i in range(ngpus): - for j in range(nvml.NVLINK_MAX_LINKS): - # By the documentation, this should be supported on PASCAL or newer, - # but this also seems to fail on newer. - with unsupported_before(handles[i], None): - cap = nvml.device_get_nvlink_capability(handles[i], j, cap_type) - assert cap >= 0 - - # Test pynvml.nvmlDeviceResetNvLinkUtilizationCounter # Test pynvml.nvmlDeviceSetNvLinkUtilizationControl # Test pynvml.nvmlDeviceGetNvLinkUtilizationCounter