Skip to content

Commit 81fb13e

Browse files
committed
Update C++ Custom Operators tutorial with note about AMD GPU.
1 parent 7262c21 commit 81fb13e

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

advanced_source/cpp_custom_ops.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -420,3 +420,7 @@ In this tutorial, we went over the recommended approach to integrating Custom C+
420420
and CUDA operators with PyTorch. The ``TORCH_LIBRARY/torch.library`` APIs are fairly
421421
low-level. For more information about how to use the API, see
422422
`The Custom Operators Manual <https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html#the-custom-operators-manual>`_.
423+
424+
Custom Operators on the AMD GPU
425+
-------------------------------
426+
This tutorial will also work on the AMD GPU with no additional modifications.

0 commit comments

Comments
 (0)