Skip to content

Commit 482fe18

Browse files
naromero77amdpytorchmergebot
authored andcommitted
Add ROCm documentation to libtorch (C++) reST. (pytorch#136378)
Fixes pytorch#126640 Added ROCm support section to libtorch (C++) reST. Pull Request resolved: pytorch#136378 Approved by: https://github.com/ezyang
1 parent 3c7edf1 commit 482fe18

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

docs/libtorch.rst

+12
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,18 @@ The core of pytorch does not depend on Python. A
55
CMake-based build system compiles the C++ source code into a shared
66
object, libtorch.so.
77

8+
AMD ROCm Support
9+
------------------------------
10+
If you're compiling for AMD ROCm then first run this command:
11+
::
12+
cd <pytorch_root>
13+
14+
# Only run this if you're compiling for ROCm
15+
python tools/amd_build/build_amd.py
16+
17+
Additional information about ROCm support can be found in the top-level
18+
`README <https://github.com/pytorch/pytorch/blob/main/README.md>`_.
19+
820
Building libtorch using Python
921
------------------------------
1022

0 commit comments

Comments
 (0)