Skip to content

Fix PyTorch FFT array-like inputs#3372

Closed
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-pytorch-fft-arraylike-20260629
Closed

Fix PyTorch FFT array-like inputs#3372
FlorianPfaff wants to merge 2 commits into
mainfrom
fix-pytorch-fft-arraylike-20260629

Conversation

@FlorianPfaff

Copy link
Copy Markdown
Owner

Summary

  • replace raw torch.fft re-exports in the PyTorch backend FFT module with thin PyRecEst-compatible wrappers
  • coerce array-like inputs before calling rfft, irfft, fftn, ifftn, fftshift, and ifftshift
  • keep NumPy-style axis/axes arguments while also accepting PyTorch-style dim
  • add regression coverage for list inputs to the PyTorch FFT helpers

Testing

  • Added tests/backend/test_pytorch_fft_array_like.py
  • Manually checked the patched wrapper behavior against local PyTorch for rfft, irfft, fftn, fftshift, and ifftshift

Full repository tests were not run in this environment.

@github-actions

Copy link
Copy Markdown
Contributor

MegaLinter analysis: Success

Descriptor Linter Files Fixed Errors Warnings Elapsed time
✅ COPYPASTE jscpd yes no no 59.23s
✅ JSON prettier 7 0 0 0 1.46s
✅ JSON v8r 7 0 0 5.2s
✅ MARKDOWN markdownlint 68 0 0 0 1.85s
✅ MARKDOWN markdown-table-formatter 68 0 0 0 0.87s
✅ PYTHON black 1095 0 0 0 55.4s
✅ PYTHON isort 1095 0 0 0 2.68s
✅ REPOSITORY checkov yes no no 44.09s
✅ REPOSITORY gitleaks yes no no 9.71s
✅ REPOSITORY git_diff yes no no 0.19s
✅ REPOSITORY secretlint yes no no 46.77s
✅ REPOSITORY syft yes no no 3.83s
✅ REPOSITORY trivy-sbom yes no no 3.52s
✅ REPOSITORY trufflehog yes no no 13.73s
✅ YAML prettier 11 0 0 0 0.66s
✅ YAML v8r 11 0 0 11.58s
✅ YAML yamllint 11 0 0 0.47s

Notices

📣 MegaLinter 9.5.0 is out! Discover the new features and security recommendations in the release announcement. (Skip this info by defining SECURITY_SUGGESTIONS: false)

See detailed reports in MegaLinter artifacts

Your project could benefit from a custom flavor, which would allow you to run only the linters you need, and thus improve runtime performances. (Skip this info by defining FLAVOR_SUGGESTIONS: false)

  • Documentation: Custom Flavors
  • Command: npx mega-linter-runner@9.5.0 --custom-flavor-setup --custom-flavor-linters PYTHON_BLACK,PYTHON_ISORT,COPYPASTE_JSCPD,JSON_V8R,JSON_PRETTIER,MARKDOWN_MARKDOWNLINT,MARKDOWN_MARKDOWN_TABLE_FORMATTER,REPOSITORY_CHECKOV,REPOSITORY_GIT_DIFF,REPOSITORY_GITLEAKS,REPOSITORY_SECRETLINT,REPOSITORY_SYFT,REPOSITORY_TRIVY_SBOM,REPOSITORY_TRUFFLEHOG,YAML_PRETTIER,YAML_YAMLLINT,YAML_V8R

MegaLinter is graciously provided by OX Security
Show us your support by starring ⭐ the repository

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant