Skip to content

Commit c3f2afd

Browse files
authored
Improve 2 NVML-related tests (NVIDIA#1542)
1 parent 4098f40 commit c3f2afd

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

cuda_bindings/tests/nvml/test_gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def test_gpu_get_platform_info(all_devices):
3535
with unsupported_before(device, None):
3636
platform_info = nvml.device_get_platform_info(device)
3737

38-
assert isinstance(platform_info, nvml.PlatformInfo_v2)
38+
assert isinstance(platform_info, (nvml.PlatformInfo_v1, nvml.PlatformInfo_v2))
3939

4040

4141
# TODO: Test APIs related to GPU instances, which require specific hardware and root

cuda_core/tests/system/test_system_device.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -640,8 +640,6 @@ def test_fan():
640640
assert isinstance(min_, int)
641641
assert isinstance(max_, int)
642642
assert min_ <= max_
643-
if speed > 0:
644-
assert min_ <= speed <= max_
645643

646644
control_policy = fan_info.control_policy
647645
assert isinstance(control_policy, system.FanControlPolicy)

0 commit comments

Comments
 (0)