From b66d07bc5817f1b1a9134359b137179089047462 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Thu, 4 Jan 2024 21:27:40 +0100
Subject: [PATCH] Add AMD doc to PR doc build

---
 .github/workflows/build_pr_documentation.yml | 13 +++++++++++++
 docs/combine_docs.py                         |  5 ++++-
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml
index e453eb179e..954e55ec48 100644
--- a/.github/workflows/build_pr_documentation.yml
+++ b/.github/workflows/build_pr_documentation.yml
@@ -64,6 +64,11 @@ jobs:
           repository: 'huggingface/optimum-furiosa'
           path: optimum-furiosa
 
+      - uses: actions/checkout@v2
+        with:
+          repository: 'huggingface/optimum-amd'
+          path: optimum-amd
+
       - name: Setup environment
         run: |
           pip uninstall -y doc-builder
@@ -118,6 +123,14 @@ jobs:
           mv furiosa-doc-build ../optimum
           cd ..
 
+      - name: Make AMD documentation
+        run: |
+          sudo docker system prune -a -f
+          cd optimum-amd
+          make doc BUILD_DIR=amd-doc-build VERSION=${{ env.VERSION }}
+          sudo mv amd-doc-build ../optimum
+          cd ..
+
       - name: Make Optimum documentation
         run: |
           sudo docker system prune -a -f
diff --git a/docs/combine_docs.py b/docs/combine_docs.py
index 2bb9bfe7a8..5424441de1 100755
--- a/docs/combine_docs.py
+++ b/docs/combine_docs.py
@@ -164,7 +164,10 @@ def main():
             # Extend table of contents sections with the subpackage name as the parent folder
             rename_subpackage_toc(subpackage, subpackage_toc)
             # Just keep the name of the partner in the TOC title
-            subpackage_toc[0]["title"] = subpackage_toc[0]["title"].split("Optimum ")[-1]
+            if subpackage == "amd":
+                subpackage_toc[0]["title"] = subpackage_toc[0]["title"].split("Optimum-")[-1]
+            else:
+                subpackage_toc[0]["title"] = subpackage_toc[0]["title"].split("Optimum ")[-1]
             if subpackage != "graphcore":
                 # Update optimum table of contents
                 base_toc.insert(1, subpackage_toc[0])