From 2dc22ca890b99410276976a262d97234966d59e1 Mon Sep 17 00:00:00 2001 From: peterjunpark Date: Wed, 7 Jan 2026 13:49:31 -0500 Subject: [PATCH] fix(primus-pytorch.rst): FP8 config instead of BF16 (#5839) --- .../rocm-for-ai/training/benchmark-docker/primus-pytorch.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst b/docs/how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst index 5323adced..2ad6129d1 100644 --- a/docs/how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst +++ b/docs/how-to/rocm-for-ai/training/benchmark-docker/primus-pytorch.rst @@ -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