Fix MIG-related tests in cuda.core.system#2065
Conversation
| def test_num_devices(): | ||
| num_devices = system.get_num_devices() | ||
| expected_num_devices = handle_return(runtime.cudaGetDeviceCount()) | ||
| assert num_devices == expected_num_devices, "Number of devices does not match expected value" |
There was a problem hiding this comment.
Just removed this test altogether since it isn't universally true and finding a way to write all of the exceptions seemed a little pointless.
This comment has been minimized.
This comment has been minimized.
| cuda_devices = get_cuda_device_names() | ||
| nvml_devices = get_nvml_device_names() | ||
|
|
||
| if any("Thor" in device["name"] for device in nvml_devices): |
There was a problem hiding this comment.
Do you know if this could also affect "Orin"?
|
|
Cursor GPT-5.4 Extra High Fast Findings
|
Fix bugs in new MIG-related functionality in cuda.core.system.