Don't use Migraphx by default

This commit is contained in:
Nicolas Mowen 2025-06-02 08:35:40 -06:00
parent fa40263742
commit 931fbc2c22

View File

@ -345,6 +345,13 @@ def get_ort_providers(
"device_type": device,
}
)
elif provider == "MIGraphXExecutionProvider":
# MIGraphX uses more CPU than ROCM, while also being the same speed
if device == "MIGraphX":
providers.append(provider)
options.append({})
else:
continue
elif provider == "CPUExecutionProvider":
providers.append(provider)
options.append(