Skip to content

Commit 2804c91

Browse files
committed
Fix tests
1 parent 1fbe63e commit 2804c91

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

cuda_core/cuda/core/system/_process.pxi

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,7 @@ class ProcessInfo:
3333
Only valid for processes running on MIG devices.
3434
"""
3535
if not self._device.mig.is_mig_device:
36-
raise nvml.NotSupportedError(
37-
"GPU instance ID is only valid for processes running on MIG devices."
38-
)
36+
raise nvml.NotSupportedError(nvml.Return.ERROR_NOT_SUPPORTED)
3937
return self._process_info.gpu_instance_id
4038

4139
@property
@@ -46,7 +44,5 @@ class ProcessInfo:
4644
Only valid for processes running on MIG devices.
4745
"""
4846
if not self._device.mig.is_mig_device:
49-
raise nvml.NotSupportedError(
50-
"Compute instance ID is only valid for processes running on MIG devices."
51-
)
47+
raise nvml.NotSupportedError(nvml.Return.ERROR_NOT_SUPPORTED)
5248
return self._process_info.compute_instance_id

0 commit comments

Comments
 (0)