microsoft / microsoft/onnxruntime

[Feature Request] Add DFT support for CUDAExecutionProvider

Open
#21,164 0 comments 1 reaction 0 assignees View on GitHub
ep:CUDA feature request
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.