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.

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

【新算子】- fft2d 算子开发 #1003

Open
PetrelYy opened this issue Apr 19, 2024 · 8 comments
Open

【新算子】- fft2d 算子开发 #1003

PetrelYy opened this issue Apr 19, 2024 · 8 comments
Assignees
Labels
ICT New Op Contribute a new operator

Comments

@PetrelYy
Copy link
Collaborator

开发计划可参考以下节点:

  1. 方案撰写,xx.xx~xx.xx
  2. 开发自测,xx.xx~xx.xx
  3. 提出 PR/MR,xx.xx~xx.xx
  4. review( 3个赞),xx.xx~xx.xx
  5. maintainer 合入
@PetrelYy PetrelYy added New Op Contribute a new operator ICT labels Apr 23, 2024
@squidruge
Copy link
Collaborator

方案pr链接:#1025

@squidruge
Copy link
Collaborator

FFT2D当前性能如图
image

@squidruge
Copy link
Collaborator

squidruge commented May 27, 2024

fft2d c2c 当前预定开发计划:

  1. 方案撰写,5.27~5.31
  2. 开发自测,6.1~6.28
  3. 提出 PR/MR,6.28~6.30
  4. review( 3个赞),7.1~7.10
  5. maintainer 合入

@squidruge
Copy link
Collaborator

ff2d.md
当前fft2d c2c方案如附件所示

@squidruge
Copy link
Collaborator

squidruge commented Jun 5, 2024

当前性能:
image

@PetrelYy
Copy link
Collaborator Author

@squidruge 还请更新进展(文档,代码完成度以及PR)及后续计划

@squidruge
Copy link
Collaborator

c2c 2d部分文档已完成,代码部分针对重点规模已完成,面向其他规模还需完善以下部分:
1)完善factor,支持64以下基的分解,预计6.19之前完成
2)恢复大质数基部分的代码,预计6.24之前完成
3)实现half版本,预计6.28之前完成

pr:#1045 (comment)

@squidruge
Copy link
Collaborator

squidruge commented Jun 21, 2024

batch =768, stride = 768 c2c 2d性能如下:
image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ICT New Op Contribute a new operator
Projects
None yet
Development

No branches or pull requests

2 participants