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, "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": elif provider == "CPUExecutionProvider":
providers.append(provider) providers.append(provider)
options.append( options.append(