Rename aimet_torch wheel file name in docs

Signed-off-by: Hitarth Mehta <quic_hitameht@quicinc.com>
quic · Jan 9, 2025 · 21337c0 · 21337c0
1 parent fae6adc
commit 21337c0
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 5 deletions.
diff --git a/Docs/install/index.rst b/Docs/install/index.rst
@@ -45,13 +45,13 @@ Use one of the following commands to install AIMET based on your choice of frame
 
         .. parsed-literal::
 
-           python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cu121\ |whl_suffix| -f |torch_pkg_url|
+           python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cu121\ |torch_whl_suffix| -f |torch_pkg_url|
 
         With CPU only:
 
         .. parsed-literal::
 
-            python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cpu\ |whl_suffix| -f |torch_pkg_url|
+            python3 -m pip install |download_url|\ |version|/aimet_torch-|version|\+cpu\ |torch_whl_suffix| -f |torch_pkg_url|
 
 
     .. tab-item:: TensorFlow
@@ -88,6 +88,7 @@ Use one of the following commands to install AIMET based on your choice of frame
 
             python3 -m pip install |download_url|\ |version|/aimet_onnx-|version|\+cpu\ |whl_suffix| -f |torch_pkg_url|
 
+.. |torch_whl_suffix| replace:: \-cp310-none-any.whl
 .. |whl_suffix| replace:: \-cp310-cp310-manylinux_2_34_x86_64.whl
 .. |download_url| replace:: \https://github.com/quic/aimet/releases/download/
 .. |torch_pkg_url| replace:: \https://download.pytorch.org/whl/torch_stable.html

diff --git a/Docs/install/install_docker.rst b/Docs/install/install_docker.rst
@@ -238,7 +238,7 @@ Releases are listed at: https://github.com/quic/aimet/releases
     export download_url="\https://github.com/quic/aimet/releases/download/${release_tag}"
 
     # Set the wheel file name with extension,
-    # for example "aimet_torch-|version|\+cu121-cp310-cp310-manylinux_2_34_x86_64.whl"
+    # for example "aimet_torch-|version|\+cu121\ |torch_whl_suffix|"
     export wheel_file_name="<wheel file name>"
 
     # NOTE: Do the following only for the PyTorch and ONNX variant packages!
@@ -266,4 +266,5 @@ Environment setup
 
     source /usr/local/lib/python3.10/dist-packages/aimet_common/bin/envsetup.sh
 
-.. _docker-install-next:
+
+.. |torch_whl_suffix| replace:: \-cp310-none-any.whl
diff --git a/Docs/install/install_host.rst b/Docs/install/install_host.rst
@@ -116,7 +116,7 @@ For example, "|version|". Releases are listed at: https://github.com/quic/aimet/
     export download_url="\https://github.com/quic/aimet/releases/download/${release_tag}"
 
     # Set the wheel file name with extension,
-    # for example "aimet_torch-|version|\+cu121-cp310-cp310-manylinux_2_34_x86_64.whl"
+    # for example "aimet_torch-|version|\+cu121\ |torch_whl_suffix|"
     export wheel_file_name="<wheel file name>"
 
     # NOTE: Do the following ONLY for the PyTorch and ONNX variant packages!
@@ -345,3 +345,6 @@ Pillow-SIMD is an optimized version of the Pillow Python Imaging Library. It can
 .. code-block:: bash
 
     source /usr/local/lib/python3.10/dist-packages/aimet_common/bin/envsetup.sh
+
+
+.. |torch_whl_suffix| replace:: \-cp310-none-any.whl