File tree 1 file changed +5
-6
lines changed
1 file changed +5
-6
lines changed Original file line number Diff line number Diff line change @@ -7,6 +7,11 @@ Custom C++ and CUDA Extensions
7
7
This tutorial is deprecated as of PyTorch 2.4. Please see :ref: `custom-ops-landing-page `
8
8
for the newest up-to-date guides on extending PyTorch with Custom C++/CUDA Extensions.
9
9
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
+
10
15
PyTorch provides a plethora of operations related to neural networks, arbitrary
11
16
tensor algebra, data wrangling and other purposes. However, you may still find
12
17
yourself in need of a more customized operation. For example, you might want to
@@ -1207,9 +1212,3 @@ examples displayed in this note `here
1207
1212
<https://github.com/pytorch/extension-cpp> `_. If you have questions, please use
1208
1213
`the forums <https://discuss.pytorch.org >`_. Also be sure to check our `FAQ
1209
1214
<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> `_.
You can’t perform that action at this time.
0 commit comments