Skip to content

Commit f7d06b6

Browse files
[ROCm] Update C++ Custom Operators tutorial with note about AMD GPU. (#3198)
* Update C++ Custom Operators tutorial with note about AMD GPU. * Put in a NOTE call out box and change to AMD ROCm. --------- Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent 5a6b217 commit f7d06b6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

advanced_source/cpp_custom_ops.rst

+4
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,10 @@ Custom C++ and CUDA Operators
1919
* PyTorch 2.4 or later
2020
* Basic understanding of C++ and CUDA programming
2121

22+
.. note::
23+
24+
This tutorial will also work on AMD ROCm with no additional modifications.
25+
2226
PyTorch offers a large library of operators that work on Tensors (e.g. torch.add, torch.sum, etc).
2327
However, you may wish to bring a new custom operator to PyTorch. This tutorial demonstrates the
2428
blessed path to authoring a custom operator written in C++/CUDA.

0 commit comments

Comments
 (0)