I want to add a c++ extension for a new device type.
I added an implementation for DeviceGuardImplInterface and added a subclass inheriting from TensorImpl for the device
I followed test/cpp_extensions/msnpu_extension.cpp but when I try to create a tensor and set the device to the new device type I get the following error:
RuntimeError: DispatchStub: unsupported device type
DispatchStub doesn’t seem to handling MSNPU or XLA devices so I was wondering how this works.
Appreciate any help