From 09965ae097bf07f1f3b693be6053ef10d2d46f23 Mon Sep 17 00:00:00 2001
From: Michael Benayoun <mickbenayoun@gmail.com>
Date: Wed, 18 Dec 2024 11:35:03 +0100
Subject: [PATCH] Fix doc

---
 docs/source/_toctree.yml                        | 17 +++++++++++++++++
 optimum/exporters/__init__.py                   |  1 +
 optimum/exporters/executorch/__init__.py        |  7 +++++--
 .../exporters/executorch/recipes/__init__.py    |  2 ++
 optimum/exporters/executorch/tasks/__init__.py  |  2 ++
 5 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/docs/source/_toctree.yml b/docs/source/_toctree.yml
index 8444da1b9a9..dc69564b045 100644
--- a/docs/source/_toctree.yml
+++ b/docs/source/_toctree.yml
@@ -81,6 +81,23 @@
       title: Reference
       isExpanded: false
     title: "ONNX"
+  - sections:
+    - local: exporters/executorch/overview
+      title: Overview
+    - sections:
+      - local: exporters/executorch/usage_guides/export_a_model
+        title: Export a model to ExecuTorch
+      - local: exporters/executorch/usage_guides/contribute
+        title: Add support for exporting an architecture to ExecuTorch
+      title: How-to guides
+    - sections:
+      - local: exporters/executorch/package_reference/configuration
+        title: ExecuTorch configurations
+      - local: exporters/executorch/package_reference/export
+        title: Export functions
+      title: Reference
+      isExpanded: false
+    title: "ExecuTorch"
   - sections:
     - local: exporters/tflite/overview
       title: Overview
diff --git a/optimum/exporters/__init__.py b/optimum/exporters/__init__.py
index eef17dac7f7..7b08812a569 100644
--- a/optimum/exporters/__init__.py
+++ b/optimum/exporters/__init__.py
@@ -13,4 +13,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from . import onnx  # noqa
+from . import executorch  # noqa
 from .tasks import TasksManager  # noqa
diff --git a/optimum/exporters/executorch/__init__.py b/optimum/exporters/executorch/__init__.py
index 67590723703..3409e69fcfb 100644
--- a/optimum/exporters/executorch/__init__.py
+++ b/optimum/exporters/executorch/__init__.py
@@ -27,8 +27,11 @@
         "discover_tasks",
         "register_task",
     ],
-    "xnnpack": [
-        "export_to_executorch_with_xnnpack",
+    "tasks": [
+        "causal_lm",
+    ],
+    "recipes": [
+        "xnnpack",
     ],
     "__main__": ["main_export"],
 }
diff --git a/optimum/exporters/executorch/recipes/__init__.py b/optimum/exporters/executorch/recipes/__init__.py
index 30466c2d1a1..a2e21cf3970 100644
--- a/optimum/exporters/executorch/recipes/__init__.py
+++ b/optimum/exporters/executorch/recipes/__init__.py
@@ -9,3 +9,5 @@
 # Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 # an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations under the License.
+
+from . import xnnpack
diff --git a/optimum/exporters/executorch/tasks/__init__.py b/optimum/exporters/executorch/tasks/__init__.py
index 30466c2d1a1..754a8241ca3 100644
--- a/optimum/exporters/executorch/tasks/__init__.py
+++ b/optimum/exporters/executorch/tasks/__init__.py
@@ -9,3 +9,5 @@
 # Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 # an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations under the License.
+
+from . import causal_lm