Getting DispatchStub error with new device

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 :slight_smile: