Skip to content

Commit 2cc4cdc

Browse files
committed
Put in a NOTE call out box and change to AMD ROCm.
1 parent bba8cb3 commit 2cc4cdc

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

advanced_source/cpp_extension.rst

+5-6
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,11 @@ Custom C++ and CUDA Extensions
77
This tutorial is deprecated as of PyTorch 2.4. Please see :ref:`custom-ops-landing-page`
88
for the newest up-to-date guides on extending PyTorch with Custom C++/CUDA Extensions.
99

10+
.. note::
11+
12+
A blog on writing extensions for AMD ROCm can be found `here
13+
<https://rocm.blogs.amd.com/artificial-intelligence/cpp-extn/readme.html>`_.
14+
1015
PyTorch provides a plethora of operations related to neural networks, arbitrary
1116
tensor algebra, data wrangling and other purposes. However, you may still find
1217
yourself in need of a more customized operation. For example, you might want to
@@ -1207,9 +1212,3 @@ examples displayed in this note `here
12071212
<https://github.com/pytorch/extension-cpp>`_. If you have questions, please use
12081213
`the forums <https://discuss.pytorch.org>`_. Also be sure to check our `FAQ
12091214
<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)