github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-01-08 22:28:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(primus-pytorch.rst): FP8 config instead of BF16 (#5839)

Browse Source
This commit is contained in:
peterjunpark
2026-01-07 13:49:31 -05:00
committed by GitHub
parent 85102079ed
commit 2dc22ca890
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst
View File

@@ -285,7 +285,7 @@ tweak some configurations (such as batch sizes).
.. code-block:: shell
EXP=examples/torchtitan/configs/MI355X/llama3.1_8B-BF16-pretrain.yaml \
EXP=examples/torchtitan/configs/MI355X/llama3.1_8B-FP8-pretrain.yaml \
bash examples/run_pretrain.sh
.. tab-item:: MI325X