mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
I gotta say, the FFT implementation is completely insane, there's gotta be a better way to do this than repeatedly inplace restriding the output tensor. Anyway, this is a faithful translation of both the MKL and cuFFT paths to Python. Fixes https://github.com/pytorch/pytorch/issues/135087 Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/145080 Approved by: https://github.com/Skylion007, https://github.com/albanD ghstack dependencies: #145530 |
||
|---|---|---|
| .. | ||
| attn_ft.py | ||
| attn_positional.py | ||
| common_utils.py | ||
| discover_coverage.py | ||
| functorch_additional_op_db.py | ||
| test_ac.py | ||
| test_ac_knapsack.py | ||
| test_aotdispatch.py | ||
| test_control_flow.py | ||
| test_dims.py | ||
| test_eager_transforms.py | ||
| test_logging.py | ||
| test_memory_efficient_fusion.py | ||
| test_minifier.py | ||
| test_ops.py | ||
| test_parsing.py | ||
| test_rearrange.py | ||
| test_vmap.py | ||
| test_vmap_registrations.py | ||
| xfail_suggester.py | ||