Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 9 additions & 1 deletion cuda_core/tests/system/test_system_device.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0


from .conftest import skip_if_nvml_unsupported, unsupported_before
from .conftest import SHOULD_SKIP_NVML_TESTS, skip_if_nvml_unsupported, unsupported_before

pytestmark = skip_if_nvml_unsupported

Expand Down Expand Up @@ -439,6 +439,10 @@ def test_module_id():
assert module_id >= 0


@pytest.mark.skipif(
not SHOULD_SKIP_NVML_TESTS and system.get_kernel_mode_driver_version()[0] < 580,
reason="nvmlDeviceGetAddressingMode requires an R580+ kernel-mode driver",
)
def test_addressing_mode():
for device in system.Device.get_all_devices():
# By docs, should be supported on TURING or newer, but experimentally,
Expand All @@ -457,6 +461,10 @@ def test_display_mode():
assert isinstance(is_display_active, bool)


@pytest.mark.skipif(
not SHOULD_SKIP_NVML_TESTS and system.get_kernel_mode_driver_version()[0] < 580,
reason="nvmlDeviceGetRepairStatus requires an R580+ kernel-mode driver",
)
def test_repair_status():
for device in system.Device.get_all_devices():
# By docs, should be supported on AMPERE or newer, but experimentally,
Expand Down
Loading