| 27 | using std::vector; |
| 28 | |
| 29 | TEST(fft2, CPP_4D) { |
| 30 | array a = randu(1024, 1024, 32); |
| 31 | array b = fft2(a); |
| 32 | |
| 33 | array A = moddims(a, 1024, 1024, 4, 8); |
| 34 | array B = fft2(A); |
| 35 | |
| 36 | cfloat *h_b = b.host<cfloat>(); |
| 37 | cfloat *h_B = B.host<cfloat>(); |
| 38 | |
| 39 | for (int i = 0; i < (int)a.elements(); i++) { |
| 40 | ASSERT_EQ(h_b[i], h_B[i]) << "at: " << i << endl; |
| 41 | } |
| 42 | |
| 43 | af_free_host(h_b); |
| 44 | af_free_host(h_B); |
| 45 | } |
| 46 | |
| 47 | TEST(ifft2, CPP_4D) { |
| 48 | array a = randu(1024, 1024, 32, c32); |