EnzymeAD / EnzymeAD/Enzyme-JAX
FFT(aA) where a is a scalar
Open
- Dominant language
- MLIR
- Stars
- 131
- Forks
- 53
- Avg merge
- 1d 10h
- Merged PRs (30d)
- 193
Description
```mlir
%129 = stablehlo.broadcast_in_dim %128, dims = [] : (tensor>) -> tensor<64x64xcomplex>
%130 = stablehlo.multiply %127, %129 : tensor<64x64xcomplex>
%131 = stablehlo.fft %130, type = FFT, length = [64, 64] {enzymexla.symmetric_matrix = [#enzymexla]} : (tensor<64x64xcomplex>) -> tensor<64x64xcomplex>
```
This is beneficial if fft output fusions, outweigh fusions on the fft operand.
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.