Skip to content

Commit

Permalink
Update doc for AMD (#1570)
Browse files Browse the repository at this point in the history
* Update index.mdx for AMD

* update doc
  • Loading branch information
mht-sharma authored Dec 6, 2023
1 parent 75ae1b7 commit c0230e0
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion docs/source/index.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ As such, Optimum enables developers to efficiently use any of these platforms wi
</a>
<a class="!no-underline border dark:border-gray-700 p-5 rounded-lg shadow hover:shadow-lg" href="./amd/index"
><div class="w-full text-center bg-gradient-to-br from-red-600 to-red-600 rounded-lg py-1.5 font-semibold mb-5 text-white text-lg leading-relaxed">AMD</div>
<p class="text-gray-700">Enable performance optimizations for <span class="underline" onclick="event.preventDefault(); window.open('https://www.amd.com/en/graphics/instinct-server-accelerators', '_blank');">AMD Instinct GPUs</span> and <span class="underline" onclick="event.preventDefault(); window.open('https://ryzenai.docs.amd.com/en/latest/index.html', '_blank');">AMD Ryzen AI IPUs</span></p>
<p class="text-gray-700">Enable performance optimizations for <span class="underline" onclick="event.preventDefault(); window.open('https://www.amd.com/en/graphics/instinct-server-accelerators', '_blank');">AMD Instinct GPUs</span> and <span class="underline" onclick="event.preventDefault(); window.open('https://ryzenai.docs.amd.com/en/latest/index.html', '_blank');">AMD Ryzen AI NPUs</span></p>
</a>
<a class="!no-underline border dark:border-gray-700 p-5 rounded-lg shadow hover:shadow-lg" href="./furiosa/index"
><div class="w-full text-center bg-gradient-to-br from-green-400 to-green-500 rounded-lg py-1.5 font-semibold mb-5 text-white text-lg leading-relaxed">FuriosaAI</div>
Expand Down
6 changes: 3 additions & 3 deletions docs/source/onnxruntime/usage_guides/amdgpu.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ pip3 install --pre torch torchvision torchaudio --index-url https://download.pyt
# Use 'rocm/pytorch:latest' as the preferred base image when using Docker for PyTorch installation.
```

### 2.2 ONNX Runtime with ROCm Execution Provider
##### 2.2 ONNX Runtime with ROCm Execution Provider

```bash
# pre-requisites
Expand Down Expand Up @@ -83,7 +83,7 @@ ValueError: Asked to use ROCMExecutionProvider as an ONNX Runtime execution prov

then something is wrong with the ROCM or ONNX Runtime installation.

### Use ROCM Execution Provider with ORT models
## Use ROCM Execution Provider with ORT models

For ORT models, the use is straightforward. Simply specify the `provider` argument in the `ORTModel.from_pretrained()` method. Here's an example:

Expand Down Expand Up @@ -128,6 +128,6 @@ Additionally, you can pass the session option `log_severity_level = 0` (verbose)
... )
```

### Observed time gains
## Observed time gains

Coming soon!

0 comments on commit c0230e0

Please sign in to comment.