Skip to content

Commit 5314416

Browse files
committed
Update C++ Extension tutorial with note about AMD GPU.
1 parent 7262c21 commit 5314416

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

advanced_source/cpp_extension.rst

+6
Original file line numberDiff line numberDiff line change
@@ -1207,3 +1207,9 @@ examples displayed in this note `here
12071207
<https://github.com/pytorch/extension-cpp>`_. If you have questions, please use
12081208
`the forums <https://discuss.pytorch.org>`_. Also be sure to check our `FAQ
12091209
<https://pytorch.org/cppdocs/notes/faq.html>`_ in case you run into any issues.
1210+
1211+
PyTorch C++ Extension on AMD GPU
1212+
--------------------------------
1213+
1214+
A blog on writing extensions for the AMD GPU can be found `here
1215+
<https://rocm.blogs.amd.com/artificial-intelligence/cpp-extn/readme.html>`_.

0 commit comments

Comments
 (0)