diff options
Diffstat (limited to 'pseudocode/operators/FFT2D.tosac')
-rw-r--r-- | pseudocode/operators/FFT2D.tosac | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/pseudocode/operators/FFT2D.tosac b/pseudocode/operators/FFT2D.tosac index a958aa4..e8f0243 100644 --- a/pseudocode/operators/FFT2D.tosac +++ b/pseudocode/operators/FFT2D.tosac @@ -22,8 +22,10 @@ for_each(0 <= n < N, 0 <= oy < H, 0 <= ox < W) { for_each(0 <= iy < H, 0 <= ix < W) { in_out_t val_real = tensor_read<in_out_t>(input_real, [N,H,W], [n,iy,ix]); in_out_t val_imag = tensor_read<in_out_t>(input_imag, [N,H,W], [n,iy,ix]); - float_t a = sign_val * 2 * pi() * ((iy * oy) / H + (ix * ox) / W); - sum_real += val_real * cos(a) + val_imag * sin(a); + int32_t ay = (static_cast<size_t>(iy) * static_cast<size_t>(oy)) % static_cast<int32_t>(H); + int32_t ax = (static_cast<size_t>(ix) * static_cast<size_t>(ox)) % static_cast<int32_t>(W); + in_out_t a = sign_val * 2 * pi() * (static_cast<in_out_t>(ay) / H + static_cast<in_out_t>(ax) / W); + sum_real += val_real * cos(a) + val_imag * sin(a); sum_imag += -val_real * sin(a) + val_imag * cos(a); } tensor_write<in_out_t>(output_real, [N,H,W], [n,oy,ox], sum_real); |