Any simple example about the new way to register custom device through PrivateUse1

frontend API
1

I know in pytorch 2.1, it is more convient for user to register their own device in pytorch,
Facilitating New Backend Integration by PrivateUse1 — PyTorch Tutorials 2.1.0+cu121 documentation,
it includes ddp amp sparce or anything else, but I have not seen any example code for how to utilize this new feature,
GitHub - Ascend/pytorch: Ascend PyTorch adapter (torch_npu). Mirror of https://gitee.com/ascend/pytorch is a good example but it is too complicated, so does the official Pytorch plan to provide a complete routine to explain how to use this new feature?

2

Hey!

I’m not sure what you mean here? The tutorial covers how to register your implementation onto the PrivateUse1 device and rename it.
Then you can use it as you use any other device. Nothing special is needed.

4

Hello, I mean any complete example instead of scattered code

5

Hey!

In that case no, I don’t think there is any plan beyond the tutorial, doc, tests and end to end example in a separate repo I’m afraid.

6

find, thanks :grinning: