-
Notifications
You must be signed in to change notification settings - Fork 536
Implement _fft_r2c core ATen op #8277
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
Conversation
Stack from ghstack (oldest at bottom): |
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/8277
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit e23ad51 with merge base 8ec08f9 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Mostly following suit on PyTorch impl here. (For your reference: https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/mkl/SpectralOps.cpp#L281) ghstack-source-id: d05d5db7f08d98baec9a37d25c4561128bd8a625 ghstack-comment-id: 2640451211 Pull Request resolved: #8277
update coming soon to fix internal build issues |
Mostly following suit on PyTorch impl here. (For your reference: https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/mkl/SpectralOps.cpp#L281) ghstack-source-id: 5caee518c2de5c48e233b442510007a66a40ee2b ghstack-comment-id: 2640451211 Pull Request resolved: #8277
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.
LGTM. Thanks @swolchok !
no code sharing for this one? |
What's to share? The interesting bits are pocketfft, and we use that, same as PyTorch core. |
I havent looked in depth but shouldnt this file kernels/optimized/cpu/op_fft_r2c.cpp be subset of https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/mkl/SpectralOps.cpp#L281 |
Mostly following suit on PyTorch impl here. (For your reference: https://github.com/pytorch/pytorch/blob/main/aten/src/ATen/native/mkl/SpectralOps.cpp#L281)