-
Notifications
You must be signed in to change notification settings - Fork 4k
Update information for ROCm 5.4 for MIGraphX and ROCm Builds #13813
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 1 commit
06bb522
3bbe68f
17e8c53
b86b621
c23f9d1
d17c536
e5511e1
a6fb305
d631e13
361a2cb
40df04b
fc925bd
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,77 @@ | ||
| --- | ||
|
Contributor
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. There is already a page for this EP here: https://onnxruntime.ai/docs/execution-providers/community-maintained/MIGraphX-ExecutionProvider.html Can/should it be consolidated?
Contributor
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I added this to mirror what was done for the ROCm EP documentation. Should we be removing the other then is what you're saying?
Contributor
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Consolidated info to this document page instead and removed the one under community-maintained. |
||
| title: AMD - MIGraphX | ||
| description: Instructions to execute ONNX Runtime with the AMD MIGraphX execution provider | ||
| parent: Execution Providers | ||
| nav_order: 10 | ||
| redirect_from: /docs/reference/execution-providers/MIGraphX-ExecutionProvider | ||
| --- | ||
|
|
||
| # ROCm Execution Provider | ||
| {: .no_toc } | ||
|
|
||
| The MIGraphX Execution Provider enables hardware accelerated computation on AMD ROCm-enabled GPUs. | ||
|
|
||
| ## Contents | ||
| {: .no_toc } | ||
|
|
||
| * TOC placeholder | ||
| {:toc} | ||
|
|
||
| ## Install | ||
|
|
||
| **NOTE** Please make sure to install the proper version of Pytorch specified here [PyTorch Version](../install/#training-install-table-for-all-languages). | ||
|
|
||
| For Nightly PyTorch builds please see [Pytorch home](https://pytorch.org/) and select ROCm as the Compute Platform. | ||
|
|
||
| Pre-built binaries of ONNX Runtime with MIGraphX EP are published for most language bindings. Please reference [Install ORT](../install). | ||
|
|
||
| ## Requirements | ||
|
|
||
|
|
||
| |ONNX Runtime|MIGraphX| | ||
| |---|---| | ||
| |main|5.4| | ||
| |1.13|5.4| | ||
| |1.13|5.3| | ||
| |1.12|5.2.3| | ||
| |1.12|5.2| | ||
|
|
||
|
|
||
| ## Build | ||
| For build instructions, please see the [BUILD page](../build/eps.md#amd-migraphx). | ||
|
|
||
| ## Usage | ||
|
|
||
| ### C/C++ | ||
|
|
||
| ```c++ | ||
| Ort::Env env = Ort::Env{ORT_LOGGING_LEVEL_ERROR, "Default"}; | ||
| Ort::SessionOptions so; | ||
| int device_id = 0; | ||
| Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_MIGraphX(so, device_id)); | ||
| ``` | ||
|
|
||
| The C API details are [here](../get-started/with-c.md). | ||
|
|
||
| ### Python | ||
| Python APIs details are [here](https://onnxruntime.ai/docs/api/python/api_summary.html). | ||
|
|
||
| ## Performance Tuning | ||
| For performance tuning, please see guidance on this page: [ONNX Runtime Perf Tuning](../performance/tune-performance.md) | ||
|
|
||
| ## Samples | ||
|
|
||
| ### Python | ||
|
|
||
| ```python | ||
| import onnxruntime as ort | ||
|
|
||
| model_path = '<path to model>' | ||
|
|
||
| providers = [ | ||
| 'MIGraphXExecutionProvider', | ||
| 'CPUExecutionProvider', | ||
| ] | ||
|
|
||
| session = ort.InferenceSession(model_path, providers=providers) | ||
| ``` | ||
Uh oh!
There was an error while loading. Please reload this page.