Add fft 1d OP

[huangyiqun]

PR Category

Operator

Type of Change

New Feature

Description

add 1d fft OP

Issue

Progress

• Change is properly reviewed (1 reviewer required, 2 recommended).
• Change is responded to an issue.
• Change is fully covered by a UT.

Performance

[tengqm]

@huangyiqun Is it possible to squash your PR into a single commit?
See this blog for a reference.

[tengqm]

Are we supposed to register this into flag_gems/__init__.py as well?

[huangyiqun]

Are we supposed to register this into flag_gems/__init__.py as well?

The speedup ratio of this operator is relatively low, and its usage frequency in mainstream models is also quite low. Therefore, I think the necessity of replacing this operator needs further discussion.

[tengqm]

Are we supposed to register this into flag_gems/__init__.py as well?

The speedup ratio of this operator is relatively low, and its usage frequency in mainstream models is also quite low. Therefore, I think the necessity of replacing this operator needs further discussion.

Thanks for the confirmation.

[bin913]

@huangyiqun Do you still keep track of this PR?