Skip to content
Draft
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
style
  • Loading branch information
echarlaix committed Sep 15, 2025
commit ba8ec01bad4988101c13e6dc99f9a8454f7043bb
12 changes: 10 additions & 2 deletions docs/combine_docs.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,11 +84,19 @@ def rename_copy_subpackage_html_paths(subpackage: str, subpackage_path: Path, op


# for docs created separately that needs to be combined
#TODO: add AMD to EXTERNAL_DOCUMENTATION once added in optimum-amd
EXTERNAL_DOCUMENTATION = {"executorch": "ExecuTorch", "habana": "Habana", "intel": "Intel", "neuron": "Neuron", "onnx":"ONNX", "tpu": "TPU"}
# TODO: add AMD to EXTERNAL_DOCUMENTATION once added in optimum-amd
EXTERNAL_DOCUMENTATION = {
"executorch": "ExecuTorch",
"habana": "Habana",
"intel": "Intel",
"neuron": "Neuron",
"onnx": "ONNX",
"tpu": "TPU",
}
# needed for cases where the section title differs from the subpackage name
TITLE_SECTION = {"Neuron": "AWS Trainium/Inferentia", "TPU": "Google TPUs"}


def add_subpackage_doc(base_toc: List, subpackage: str):
"""
Extends the table of content with a section generated by the original optimum-xxx.
Expand Down