From 482fe186b9e945aac598e9d10d4eb39f6b49d0c2 Mon Sep 17 00:00:00 2001
From: "Nichols A. Romero" <nick.romero@amd.com>
Date: Wed, 25 Sep 2024 02:30:54 +0000
Subject: [PATCH] Add ROCm documentation to libtorch (C++) reST. (#136378)

Fixes #126640

Added ROCm support section to libtorch (C++) reST.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/136378
Approved by: https://github.com/ezyang
---
 docs/libtorch.rst | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/docs/libtorch.rst b/docs/libtorch.rst
index 2e638ed87c80..f4a678caab12 100644
--- a/docs/libtorch.rst
+++ b/docs/libtorch.rst
@@ -5,6 +5,18 @@ The core of pytorch does not depend on Python. A
 CMake-based build system compiles the C++ source code into a shared
 object, libtorch.so.
 
+AMD ROCm Support
+------------------------------
+If you're compiling for AMD ROCm then first run this command:
+::
+   cd <pytorch_root>
+
+   # Only run this if you're compiling for ROCm
+   python tools/amd_build/build_amd.py
+
+Additional information about ROCm support can be found in the top-level
+`README <https://github.com/pytorch/pytorch/blob/main/README.md>`_.
+
 Building libtorch using Python
 ------------------------------