Metadata-Version: 2.4
Name: onnxruntime-ep-migraphx
Version: 1.0.0+rocm10.1.0
Summary: ONNX Runtime MIGraphX Execution Provider (requires AMD ROCm and MIGraphX -- see Installation)
Author: AMD
License: MIT
Project-URL: Repository, https://github.com/onnxruntime/onnxruntime-ep-amdgpu
Classifier: Development Status :: 3 - Alpha
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.9
Requires-Dist: onnxruntime~=1.29.0
Requires-Dist: migraphx-libs==2.18.0+rocm10.1.0
Provides-Extra: device-gfx908
Requires-Dist: rocm[libraries,device-gfx908]==10.1.0; extra == "device-gfx908"
Provides-Extra: device-gfx90a
Requires-Dist: rocm[libraries,device-gfx90a]==10.1.0; extra == "device-gfx90a"
Provides-Extra: device-gfx942
Requires-Dist: rocm[libraries,device-gfx942]==10.1.0; extra == "device-gfx942"
Provides-Extra: device-gfx950
Requires-Dist: rocm[libraries,device-gfx950]==10.1.0; extra == "device-gfx950"
Provides-Extra: device-gfx1010
Requires-Dist: rocm[libraries,device-gfx1010]==10.1.0; extra == "device-gfx1010"
Provides-Extra: device-gfx1011
Requires-Dist: rocm[libraries,device-gfx1011]==10.1.0; extra == "device-gfx1011"
Provides-Extra: device-gfx1012
Requires-Dist: rocm[libraries,device-gfx1012]==10.1.0; extra == "device-gfx1012"
Provides-Extra: device-gfx1030
Requires-Dist: rocm[libraries,device-gfx1030]==10.1.0; extra == "device-gfx1030"
Provides-Extra: device-gfx1031
Requires-Dist: rocm[libraries,device-gfx1031]==10.1.0; extra == "device-gfx1031"
Provides-Extra: device-gfx1032
Requires-Dist: rocm[libraries,device-gfx1032]==10.1.0; extra == "device-gfx1032"
Provides-Extra: device-gfx1033
Requires-Dist: rocm[libraries,device-gfx1033]==10.1.0; extra == "device-gfx1033"
Provides-Extra: device-gfx1034
Requires-Dist: rocm[libraries,device-gfx1034]==10.1.0; extra == "device-gfx1034"
Provides-Extra: device-gfx1035
Requires-Dist: rocm[libraries,device-gfx1035]==10.1.0; extra == "device-gfx1035"
Provides-Extra: device-gfx1036
Requires-Dist: rocm[libraries,device-gfx1036]==10.1.0; extra == "device-gfx1036"
Provides-Extra: device-gfx1100
Requires-Dist: rocm[libraries,device-gfx1100]==10.1.0; extra == "device-gfx1100"
Provides-Extra: device-gfx1101
Requires-Dist: rocm[libraries,device-gfx1101]==10.1.0; extra == "device-gfx1101"
Provides-Extra: device-gfx1102
Requires-Dist: rocm[libraries,device-gfx1102]==10.1.0; extra == "device-gfx1102"
Provides-Extra: device-gfx1103
Requires-Dist: rocm[libraries,device-gfx1103]==10.1.0; extra == "device-gfx1103"
Provides-Extra: device-gfx1150
Requires-Dist: rocm[libraries,device-gfx1150]==10.1.0; extra == "device-gfx1150"
Provides-Extra: device-gfx1151
Requires-Dist: rocm[libraries,device-gfx1151]==10.1.0; extra == "device-gfx1151"
Provides-Extra: device-gfx1152
Requires-Dist: rocm[libraries,device-gfx1152]==10.1.0; extra == "device-gfx1152"
Provides-Extra: device-gfx1153
Requires-Dist: rocm[libraries,device-gfx1153]==10.1.0; extra == "device-gfx1153"
Provides-Extra: device-gfx1200
Requires-Dist: rocm[libraries,device-gfx1200]==10.1.0; extra == "device-gfx1200"
Provides-Extra: device-gfx1201
Requires-Dist: rocm[libraries,device-gfx1201]==10.1.0; extra == "device-gfx1201"
Provides-Extra: device-gfx1250
Requires-Dist: rocm[libraries,device-gfx1250]==10.1.0; extra == "device-gfx1250"
Provides-Extra: device-all
Requires-Dist: rocm[libraries,device-all]==10.1.0; extra == "device-all"
Description-Content-Type: text/markdown

# onnxruntime-ep-migraphx

MIGraphX execution provider plugin for ONNX Runtime, providing GPU-accelerated
inference on AMD hardware.

Source: https://github.com/onnxruntime/onnxruntime-ep-amdgpu

## Installation

Select the `device-<target>` extra matching your GPU to install ROCm 10.1.0
alongside the plugin:

```
pip install onnxruntime-ep-migraphx[device-gfx942] \
    --extra-index-url https://stable.repo.amd.com/rocm/migraphx/whl-next \
    --extra-index-url https://stable.repo.amd.com/rocm/whl-next
```

`device-all` covers every supported GPU at the cost of a much larger download.
If ROCm is already installed, the `device-<target>` extra can be omitted.

The two extra indexes supply `migraphx-libs` and
`rocm[libraries,device-<target>]==10.1.0` respectively; `onnxruntime` still
resolves from PyPI. Either `--extra-index-url` may be omitted if the package
is already installed or once it becomes available on PyPI.

To determine the right target:

```
$ pip install rocm-bootstrap
$ rocm-bootstrap-detect
gfx1103
```

## Usage

```python
import onnxruntime as ort
import onnxruntime_ep_migraphx as m

# Register the execution provider
for name, path in zip(m.get_ep_names(), m.get_library_paths()):
    ort.register_execution_provider_library(name, path)

print(ort.get_available_providers())
# ['AzureExecutionProvider', 'CPUExecutionProvider', 'MIGraphXExecutionProvider']

# List available devices
devices = [d for d in ort.get_ep_devices() if d.ep_name == m.get_ep_name()]
for d in devices:
    print(d.ep_name, d.device.type)
```
