Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature Request] Add DFT support for CUDAExecutionProvider #21164

Open
alexander-veselov opened this issue Jun 25, 2024 · 0 comments
Open

[Feature Request] Add DFT support for CUDAExecutionProvider #21164

alexander-veselov opened this issue Jun 25, 2024 · 0 comments
Labels
ep:CUDA issues related to the CUDA execution provider feature request request for unsupported feature or enhancement

Comments

@alexander-veselov
Copy link

Describe the feature request

While running ONNX model that contains DFT 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 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.

@alexander-veselov alexander-veselov added the feature request request for unsupported feature or enhancement label Jun 25, 2024
@github-actions github-actions bot added the ep:CUDA issues related to the CUDA execution provider label Jun 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ep:CUDA issues related to the CUDA execution provider feature request request for unsupported feature or enhancement
Projects
None yet
Development

No branches or pull requests

1 participant