Put in a NOTE call out box and change to AMD ROCm.

Author: naromero77amd

advanced_source/cpp_custom_ops.rst

@@ -19,6 +19,10 @@ Custom C++ and CUDA Operators
        * PyTorch 2.4 or later
        * Basic understanding of C++ and CUDA programming
 
+.. note::
+
+  This tutorial will also work on AMD ROCm with no additional modifications.
+
 PyTorch offers a large library of operators that work on Tensors (e.g. torch.add, torch.sum, etc).
 However, you may wish to bring a new custom operator to PyTorch. This tutorial demonstrates the
 blessed path to authoring a custom operator written in C++/CUDA.
@@ -420,7 +424,3 @@ In this tutorial, we went over the recommended approach to integrating Custom C+
 and CUDA operators with PyTorch. The ``TORCH_LIBRARY/torch.library`` APIs are fairly
 low-level. For more information about how to use the API, see
 `The Custom Operators Manual <https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html#the-custom-operators-manual>`_.
-
-Custom Operators on the AMD GPU
--------------------------------
-This tutorial will also work on the AMD GPU with no additional modifications.