microsoft / microsoft/onnxruntime
[Feature Request] Add DFT support for CUDAExecutionProvider
- Dominant language
- C++
- Stars
- 21.9k
- Forks
- 4.2k
- Avg merge
- 4d 11h
- Merged PRs (30d)
- 184
Description
### Describe the feature request
While running ONNX model that contains [DFT](https://onnx.ai/onnx/operators/onnx__DFT.html) operation using CUDAExecutionProvider, I see messages like: "CUDA kernel not found in registries for Op type: DFT".
Is DFT supported while using CUDAExecutionProvider? Or is this something with my model?
DFT is performance bottleneck for my model, so it would be really nice to have CUDA support for DFT operation.
There is [NVidia cuFFT](https://docs.nvidia.com/cuda/cufft/) library. Seems to be exactly what I need.
1. Can I expect that DFT operation will be implemented for CUDAExecutionProvider in near future?
2. Can I workaround my problem somehow to run DFT on GPU?
### Describe scenario use case
Better performance for DFT operation when CUDAExecutionProvider is available.
Contributor guide
Research direction
The issue names no repository files or tests. Start by locating CUDAExecutionProvider's handling of the ONNX DFT operation and compare it with the linked cuFFT documentation. Done would require validated GPU support for DFT and improved performance for the reported model.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- backend, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100