【Hackathon 9th No.114】torch._C._fft.fft_rfft API转换 torch.fft.rfft #331
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
Thanks for your contribution! |
Contributor
Author
|
lixinqi
reviewed
Nov 6, 2025
| if self.unstable_api == "torch._C._nn.avg_pool2d": | ||
| gm = self.avg_pool2d_to_avg_pool2d(gm) | ||
| elif self.unstable_api == "torch._C._fft.fft_rfft": | ||
| gm = self.fft_rfft_to_rfft(gm) |
Collaborator
There was a problem hiding this comment.
这里每次提交都会冲突，给合入造成了一些小困扰。我建立在这里写一行这样的代码：
methods = (name for name in vars(type(self)).keys() if name.startswith('_impl_unstable_to_stable'))
for method in methods:
gm = getattr(self, method)(gm)这样每个人的代码就写到各自的_impl_unstable_to_stable_XXX方法里，不再彼此干扰。
lixinqi
approved these changes
Nov 6, 2025
Collaborator
|
commit log尽量写英文 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
为了解决 GraphNet 计算图中使用不稳定 API torch._C._fft.fft_rfft 导致无法转换为 Paddle 计算图的问题，新增了 fft_rfft_to_rfft 方法。该方法会遍历 FX Graph 的所有节点，查找 target 为 torch._C._fft.fft_rfft 的节点，并将其替换为 PyTorch 官方稳定 API torch.fft.rfft。替换完成后，重新编译计算图，确保后续模型运行时只会调用稳定接口。