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
DFT on DirectML #12710
DFT on DirectML #12710
Conversation
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/GeneratedShaders/gen.bat
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good to me.
This pull request fixes 5 alerts when merging ed027de into a972db0 - view on LGTM.com fixed alerts:
|
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlDFT.h
Show resolved
Hide resolved
|
||
uint3 idx = uint3(0, 0, 0); | ||
idx.x = index / (OutputSizes[1] * OutputSizes[2]); | ||
idx.y = temp / OutputSizes[2]; // This corresponds to the s1'th element of the dft |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/Shaders/stockham.hlsl
Outdated
Show resolved
Hide resolved
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/Shaders/stockham.hlsl
Show resolved
Hide resolved
Thanks @smk2007 for the work. It'll be super exciting to see this in Premier Pro audio scenarios at MAX this year. We should find a way to integrate this shader into DML in its next quarterly update. |
* DFT on DirectML * feedback * fix misc build issues * fixes * fix constant cpu inputs and optional tensors for external operators * disable dft tests on 'pure' dml
Add kernel implementation for DFT on DirectML.