From e8d104124f3d94d12812f479922971ee1a012cfd Mon Sep 17 00:00:00 2001
From: Peter Park <peter.park@amd.com>
Date: Tue, 16 Sep 2025 17:21:57 -0400
Subject: [PATCH] Fix PyTorch training benchmark doc template (#5357)

* fix template

* update wordlist
---
 .wordlist.txt                                 |  3 ++
 .../benchmark-docker/pytorch-training.rst     | 30 ++++++++++---------
 2 files changed, 19 insertions(+), 14 deletions(-)

diff --git a/.wordlist.txt b/.wordlist.txt
index ad946483b..982320da9 100644
--- a/.wordlist.txt
+++ b/.wordlist.txt
@@ -622,6 +622,7 @@ coalescable
 codename
 collater
 comgr
+compat
 completers
 composable
 concretization
@@ -783,6 +784,7 @@ lossy
 macOS
 matchers
 maxtext
+megablocks
 megatron
 microarchitecture
 migraphx
@@ -941,6 +943,7 @@ softmax
 spack
 spmm
 src
+stanford
 stochastically
 strided
 subcommand
diff --git a/docs/how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst b/docs/how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst
index d52293d39..f2e52fc65 100644
--- a/docs/how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst
+++ b/docs/how-to/rocm-for-ai/training/benchmark-docker/pytorch-training.rst
@@ -164,20 +164,20 @@ Run training
    {% for model_group in model_groups %}
       {% for model in model_group.models %}
 
-         The following run command is tailored to {{ model.model }}.
-         See :ref:`amd-pytorch-training-model-support` to switch to another available model.
-
-         1. Clone the ROCm Model Automation and Dashboarding (`<https://github.com/ROCm/MAD>`__) repository to a local
-            directory and install the required packages on the host machine.
-
-            .. code-block:: shell
-
-               git clone https://github.com/ROCm/MAD
-               cd MAD
-               pip install -r requirements.txt
-
          .. container:: model-doc {{ model.mad_tag }}
 
+            The following run command is tailored to {{ model.model }}.
+            See :ref:`amd-pytorch-training-model-support` to switch to another available model.
+
+            1. Clone the ROCm Model Automation and Dashboarding (`<https://github.com/ROCm/MAD>`__) repository to a local
+               directory and install the required packages on the host machine.
+
+               .. code-block:: shell
+
+                  git clone https://github.com/ROCm/MAD
+                  cd MAD
+                  pip install -r requirements.txt
+
             2. For example, use this command to run the performance benchmark test on the {{ model.model }} model
                using one node with the {{ model.precision }} data type on the host machine.
 
@@ -202,8 +202,10 @@ Run training
    {% for model_group in model_groups %}
       {% for model in model_group.models %}
 
-         The following commands are tailored to {{ model.model }}.
-         See :ref:`amd-pytorch-training-model-support` to switch to another available model.
+         .. container:: model-doc {{ model.mad_tag }}
+
+            The following commands are tailored to {{ model.model }}.
+            See :ref:`amd-pytorch-training-model-support` to switch to another available model.
 
       {% endfor %}
    {% endfor %}