Merge branch 'main' into svekars-patch-37

Author: svekars

advanced_source/cpp_extension.rst

@@ -1207,3 +1207,5 @@ examples displayed in this note `here
 <https://github.com/pytorch/extension-cpp>`_. If you have questions, please use
 `the forums <https://discuss.pytorch.org>`_. Also be sure to check our `FAQ
 <https://pytorch.org/cppdocs/notes/faq.html>`_ in case you run into any issues.
+A blog on writing extensions for AMD ROCm can be found `here
+<https://rocm.blogs.amd.com/artificial-intelligence/cpp-extn/readme.html>`_.

advanced_source/python_custom_ops.py

@@ -112,7 +112,10 @@ def crop(pic: torch.Tensor, box: Sequence[int]) -> torch.Tensor:
 def _(pic, box):
     channels = pic.shape[0]
     x0, y0, x1, y1 = box
-    return pic.new_empty(channels, y1 - y0, x1 - x0)
+    result = pic.new_empty(y1 - y0, x1 - x0, channels).permute(2, 0, 1)
+    # The result should have the same metadata (shape/strides/``dtype``/device)
+    # as running the ``crop`` function above.
+    return result
 
 ######################################################################
 # After this, ``crop`` now works without graph breaks: