pytorch_open_registration_example

pytorch_open_registration_example copied to clipboard

I have run the code successfully in this repository, but when I run the code below: ```py import torch from utils.custom_device_mode import foo_module, enable_foo_device a = torch.randn(4, device='privateuseone') ``` I...

andakai

how to run this example?

2

rt, please help

guoxiangzhou

I notices in the code you use `register_privateuse1_backend('foo')` while in nighly it didn't work but `torch.utils.rename_privateuse1_backend('foo')` this works. Is it going to be the future API? I've updated my OpenCL...

artyom-beilis

Several issues 1. issues with changed const and missing `copy_data` from Allocator class 2. `torch.register_privateuse1_backend` -> `+torch.utils.rename_privateuse1_backend` However even with all that I still get failure during runtime: ``` Traceback...

artyom-beilis