From d04443ac13ef829af2507374c540a26e50331ad4 Mon Sep 17 00:00:00 2001
From: Peter Park <peter.park@amd.com>
Date: Thu, 24 Apr 2025 15:48:53 -0400
Subject: [PATCH] Add note for chai-1 benchmark Docker in
 pytorch-inference-benchmark.rst (#4684)

(cherry picked from commit 1f41ce26be5aceeb216cd10aada5c321fa8d42db)
---
 .../rocm-for-ai/inference/pytorch-inference-benchmark.rst     | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst b/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst
index 3cf8bca03..2e5d50d27 100644
--- a/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst
+++ b/docs/how-to/rocm-for-ai/inference/pytorch-inference-benchmark.rst
@@ -92,6 +92,10 @@ PyTorch inference performance testing
 
             docker pull rocm/pytorch:rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0_triton_llvm_reg_issue
 
+         .. note::
+
+            The Chai-1 benchmark uses a specifically selected Docker image using ROCm 6.2.3 and PyTorch 2.3.0 to address an accuracy issue.
+
    .. container:: model-doc pyt_clip_inference
 
       2. Use the following command to pull the `ROCm PyTorch Docker image <https://hub.docker.com/layers/rocm/pytorch/rocm6.2.3_ubuntu22.04_py3.10_pytorch_release_2.3.0_triton_llvm_reg_issue/images/sha256-b736a4239ab38a9d0e448af6d4adca83b117debed00bfbe33846f99c4540f79b>`_ from Docker Hub.