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.

Sign up for GitHub

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

Jump to bottom

[torchlib] Fix _fft_c2r #1844

Open
wants to merge 19 commits into
base: main
Choose a base branch
from
Open

[torchlib] Fix _fft_c2r #1844

wants to merge 19 commits into from

Conversation

justinchuby
Copy link
Collaborator

@justinchuby justinchuby commented Sep 3, 2024 •
edited
Loading

@codecov Codecov
Copy link

codecov bot commented Sep 3, 2024 •
edited
Loading

Codecov Report

Attention: Patch coverage is 10.81081% with 33 lines in your changes missing coverage. Please review.

Project coverage is 73.63%. Comparing base (84dfcad) to head (26779c9).
Report is 2 commits behind head on main.

Files with missing lines Patch % Lines
onnxscript/function_libs/torch_lib/ops/fft.py 10.81% 33 Missing ⚠️
Additional details and impacted files 
@@            Coverage Diff             @@
##             main    #1844      +/-   ##
==========================================
- Coverage   73.77%   73.63%   -0.14%     
==========================================
  Files         225      225              
  Lines       29333    29343      +10     
  Branches     3467     3470       +3     
==========================================
- Hits        21639    21606      -33     
- Misses       6560     6601      +41     
- Partials     1134     1136       +2

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@justinchuby
Copy link
Collaborator Author

https://github.com/pytorch/pytorch/blob/a00fad017719346bac6e08da0819358146e647e3/torch/_refs/fft.py#L134

@justinchuby
Copy link
Collaborator Author

r2c needs conjugation: https://github.com/pytorch/pytorch/blob/a00fad017719346bac6e08da0819358146e647e3/torch/_refs/fft.py#L169

@justinchuby
Copy link
Collaborator Author

Just specify dft_length

github-advanced-security[bot]
github-advanced-security bot found potential problems Sep 3, 2024
View reviewed changes
onnxscript/function_libs/torch_lib/ops/fft.py
@@ -1,7 +1,5 @@
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Copyright (c) Microsoft Corporation.

Check warning

Code scanning / lintrunner

RUFF-FORMAT/format Warning

Run lintrunner -a to apply this patch.
github-advanced-security[bot]
github-advanced-security bot found potential problems Sep 3, 2024
View reviewed changes
onnxscript/function_libs/torch_lib/ops/fft.py
@@ -1,7 +1,5 @@
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Copyright (c) Microsoft Corporation.

Check warning

Code scanning / lintrunner

RUFF/format Warning

Run lintrunner -a to apply this patch.
onnxscript/function_libs/torch_lib/ops/fft.py
@@ -12,7 +10,7 @@

from __future__ import annotations

from typing import Optional, Sequence
from typing import Literal, Optional, Sequence

Check warning

Code scanning / lintrunner

PYLINT/W0611 Warning

Unused Literal imported from typing (unused-import)
See unused-import. To disable, use # pylint: disable=unused-import
onnxscript/function_libs/torch_lib/ops/fft.py
@@ -12,7 +10,7 @@

from __future__ import annotations

from typing import Optional, Sequence
from typing import Literal, Optional, Sequence

Check warning

Code scanning / lintrunner

RUFF/F401 Warning

typing.Literal imported but unused.
See https://docs.astral.sh/ruff/rules/unused-import
onnxscript/function_libs/torch_lib/ops/fft.py
dims: Sequence[int],
forward: bool,

Check warning

Code scanning / lintrunner

PYLINT/W0613 Warning

Unused argument 'forward' (unused-argument)
See unused-argument. To disable, use # pylint: disable=unused-argument
onnxscript/function_libs/torch_lib/ops/fft.py
result = transformed
transformed = self

signal_size = _compute_signal_size(self, dims, last_dim_size)

Check warning

Code scanning / lintrunner

PYLINT/W0612 Warning

Unused variable 'signal_size' (unused-variable)
See unused-variable. To disable, use # pylint: disable=unused-variable
onnxscript/function_libs/torch_lib/ops/fft.py
# If normalization is 1/n and we are in backward mode, we use the inverse
# mode in ONNX to get the 1/n normalization.
inverse = normalization == 2 and not forward
ortho = normalization == 1

Check warning

Code scanning / lintrunner

RUFF/F841 Warning

Local variable ortho is assigned to but never used.
See https://docs.astral.sh/ruff/rules/unused-variable
onnxscript/function_libs/torch_lib/ops/fft.py

# Remove the batch dimension
transformed = op.Squeeze(transformed, axes=[0])
normalized = _fftn_onnx_normalization(

Check failure

Code scanning / lintrunner

PYLINT/E0602 Error

Undefined variable '_fftn_onnx_normalization' (undefined-variable)
See undefined-variable. To disable, use # pylint: disable=undefined-variable
onnxscript/function_libs/torch_lib/ops/fft.py

# Remove the batch dimension
transformed = op.Squeeze(transformed, axes=[0])
normalized = _fftn_onnx_normalization(

Check failure

Code scanning / lintrunner

RUFF/F821 Error

Undefined name \_fftn\_onnx\_normalization.
See https://docs.astral.sh/ruff/rules/undefined-name
onnxscript/function_libs/torch_lib/ops/fft.py
self_rank = len(self.shape)
# ONNX DFT input assumes the last dimension is the complex dimension.
# Thus dim=-1 in PyTorch is dim=-2 in ONNX.
dim = [(d - 1) + self_rank if d < 0 else d for d in dim]
transformed = _fftn_onnx(self, dim, normalization, inverse=True, onesided=False)
transformed = _fftn_onnx(

Check failure

Code scanning / lintrunner

PYLINT/E1123 Error

Unexpected keyword argument 'inverse' in function call (unexpected-keyword-arg)
See unexpected-keyword-arg. To disable, use # pylint: disable=unexpected-keyword-arg
onnxscript/function_libs/torch_lib/ops/fft.py
self_rank = len(self.shape)
# ONNX DFT input assumes the last dimension is the complex dimension.
# Thus dim=-1 in PyTorch is dim=-2 in ONNX.
dim = [(d - 1) + self_rank if d < 0 else d for d in dim]
transformed = _fftn_onnx(self, dim, normalization, inverse=True, onesided=False)
transformed = _fftn_onnx(

Check failure

Code scanning / lintrunner

PYLINT/E1120 Error

No value for argument 'forward' in function call (no-value-for-parameter)
See no-value-for-parameter. To disable, use # pylint: disable=no-value-for-parameter
This was referenced Sep 4, 2024
Open
Open
@justinchuby justinchuby self-assigned this Jan 27, 2025
@justinchuby justinchuby marked this pull request as ready for review January 27, 2025 20:20
@justinchuby justinchuby added the module: torchlib Related to the torch/aten function lib in development label Jan 27, 2025
github-advanced-security[bot]
github-advanced-security bot found potential problems Jan 27, 2025
View reviewed changes
onnxscript/function_libs/torch_lib/ops/fft.py
@@ -12,7 +10,7 @@

from __future__ import annotations

from typing import Optional, Sequence
from typing import Literal, Optional, Sequence

Check notice

Code scanning / CodeQL

Unused import Note

Import of 'Literal' is not used.
onnxscript/function_libs/torch_lib/ops/fft.py
result = transformed
transformed = self

signal_size = _compute_signal_size(self, dims, last_dim_size)

Check notice

Code scanning / CodeQL

Unused local variable Note

Variable signal_size is not used.
onnxscript/function_libs/torch_lib/ops/fft.py

# Torch computes one-sided FFT on the last dimension only.
if onesided:
transformed = op.DFT(transformed, axis=dims[-1], onesided=True)

Check notice

Code scanning / CodeQL

Unused local variable Note

Variable transformed is not used.
onnxscript/function_libs/torch_lib/ops/fft.py
transformed = op.DFT(transformed, axis=dims[-1], onesided=True)
# TODO: Update signal_size for one-sided FFT
elif last_dim_size is not None:
transformed = op.DFT(

Check notice

Code scanning / CodeQL

Unused local variable Note

Variable transformed is not used.
onnxscript/function_libs/torch_lib/ops/fft.py
result = transformed
else:
result = op.Mul(transformed, total_sample_count)
transformed = op.DFT(transformed, axis=dims[-1], onesided=False)

Check notice

Code scanning / CodeQL

Unused local variable Note

Variable transformed is not used.
onnxscript/function_libs/torch_lib/ops/fft.py
# If normalization is 1/n and we are in backward mode, we use the inverse
# mode in ONNX to get the 1/n normalization.
inverse = normalization == 2 and not forward
ortho = normalization == 1

Check notice

Code scanning / CodeQL

Unused local variable Note

Variable ortho is not used.
onnxscript/function_libs/torch_lib/ops/fft.py
Comment on lines +172 to +174
transformed = _fftn_onnx(
self, dim, normalization, inverse=True, onesided=False, last_dim_size=last_dim_size
)

Check failure

Code scanning / CodeQL

Wrong name for an argument in a call Error

Keyword argument 'inverse' is not a supported parameter name of
function _fftn_onnx
Loading
.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@justinchuby justinchuby

Labels
module: torchlib Related to the torch/aten function lib in development
Projects
ONNX Script Review Board
Status: Todo
Milestone
No milestone
2 participants
@justinchuby @fatcat-z