From e0927976d06d163ed09fe5bd80d013e1cfa0c463 Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Mon, 4 Dec 2023 15:06:08 +0100 Subject: [PATCH] Add amd documentation (#1557) * add install instructions * add link * add links * Update README.md * Update docs/source/installation.mdx --------- Co-authored-by: fxmarty <9808326+fxmarty@users.noreply.github.com> --- README.md | 1 + docs/source/index.mdx | 6 +++--- docs/source/installation.mdx | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index c86fa19c5a6..d18cc041012 100644 --- a/README.md +++ b/README.md @@ -19,6 +19,7 @@ If you'd like to use the accelerator-specific features of 🤗 Optimum, you can | [ONNX Runtime](https://onnxruntime.ai/docs/) | `pip install --upgrade-strategy eager optimum[onnxruntime]` | | [Intel Neural Compressor](https://www.intel.com/content/www/us/en/developer/tools/oneapi/neural-compressor.html) | `pip install --upgrade-strategy eager optimum[neural-compressor]`| | [OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager optimum[openvino,nncf]` | +| [AMD Instinct GPUs and Ryzen AI NPU](https://www.amd.com/en/graphics/instinct-server-accelerators) | `pip install --upgrade-strategy eager optimum[amd]` | | [Habana Gaudi Processor (HPU)](https://habana.ai/training/) | `pip install --upgrade-strategy eager optimum[habana]` | | [FuriosaAI](https://www.furiosa.ai/) | `pip install --upgrade-strategy eager optimum[furiosa]` | diff --git a/docs/source/index.mdx b/docs/source/index.mdx index 0c9a9c87f1e..d3c7399a4c7 100644 --- a/docs/source/index.mdx +++ b/docs/source/index.mdx @@ -33,9 +33,9 @@ As such, Optimum enables developers to efficiently use any of these platforms wi >
AWS Trainium/Inferentia

Accelerate your training and inference workflows with AWS Trainium and AWS Inferentia

-
AMD Instinct GPUs
-

Available soon AMD Instinct GPUs

+
AMD
+

Enable performance optimizations for AMD Instinct GPUs and AMD Ryzen AI IPUs

FuriosaAI
diff --git a/docs/source/installation.mdx b/docs/source/installation.mdx index d8a41d973fb..cf84d429ea6 100644 --- a/docs/source/installation.mdx +++ b/docs/source/installation.mdx @@ -25,7 +25,7 @@ If you'd like to use the accelerator-specific features of 🤗 Optimum, you can | [ONNX runtime](https://onnxruntime.ai/docs/) | `pip install --upgrade-strategy eager install optimum[onnxruntime]`| | [Intel Neural Compressor (INC)](https://www.intel.com/content/www/us/en/developer/tools/oneapi/neural-compressor.html) | `pip install --upgrade-strategy eager optimum[neural-compressor]` | | [Intel OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager optimum[openvino,nncf]` | -| [AMD Instinct GPUs](https://www.amd.com/en/graphics/instinct-server-accelerators) | Available soon | +| [AMD Instinct GPUs and Ryzen AI NPU](https://www.amd.com/en/graphics/instinct-server-accelerators) | `pip install --upgrade-strategy eager optimum[amd]` | | [Habana Gaudi Processor (HPU)](https://habana.ai/training/) | `pip install --upgrade-strategy eager optimum[habana]` | | [FuriosaAI](https://www.furiosa.ai/) | `pip install --upgrade-strategy eager optimum[furiosa]` |