microsoft / tensorflow-directml-plugin

DirectML PluggableDevice plugin for TensorFlow 2
Apache License 2.0
179 stars 23 forks source link

Adding a new device #363

Closed RoySegal closed 8 months ago

RoySegal commented 1 year ago

Hi, I am intreseted to find out what I need to do to add a new accelerator under DirectML. I noticed that it currently supports Direct3D (DirectX12 devices) - which means GPUs only. However, this accelerator isn't related to display and doesn't support DirectX12 framework. What is the roadmap regarding adding support to non-display (non-GPU) hardware for DirectML? Is there any way to integrate my accelerator into DirectML at the moment?

Thanks, Roy

PatriceVignola commented 8 months ago

DirectML only supports DirectX 12 devices, and there aren't any plans to support other accelerators.