diff --git a/unsupported/test/CMakeLists.txt b/unsupported/test/CMakeLists.txt index 5a9ed5730..3d82508f7 100644 --- a/unsupported/test/CMakeLists.txt +++ b/unsupported/test/CMakeLists.txt @@ -144,6 +144,8 @@ if(EIGEN_TEST_CXX11) ei_add_test(cxx11_tensor_custom_op "-std=c++0x") ei_add_test(cxx11_tensor_custom_index "-std=c++0x") ei_add_test(cxx11_tensor_sugar "-std=c++0x") + ei_add_test(cxx11_tensor_fft "-std=c++0x") + ei_add_test(cxx11_tensor_ifft "-std=c++0x") # These tests needs nvcc # ei_add_test(cxx11_tensor_device "-std=c++0x") diff --git a/unsupported/test/cxx11_tensor_fft.cpp b/unsupported/test/cxx11_tensor_fft.cpp new file mode 100644 index 000000000..4aefcc79c --- /dev/null +++ b/unsupported/test/cxx11_tensor_fft.cpp @@ -0,0 +1,273 @@ +// This file is part of Eigen, a lightweight C++ template library +// for linear algebra. +// +// Copyright (C) 2014 Jianwei Cui +// +// This Source Code Form is subject to the terms of the Mozilla +// Public License v. 2.0. If a copy of the MPL was not distributed +// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. + +#include "main.h" +#include + +using Eigen::Tensor; + +template +static void test_fft_2D_golden() { + Tensor input(2, 3); + input(0, 0) = 1; + input(0, 1) = 2; + input(0, 2) = 3; + input(1, 0) = 4; + input(1, 1) = 5; + input(1, 2) = 6; + + array fft; + fft[0] = 0; + fft[1] = 1; + + Tensor, 2, DataLayout, long> output = input.template fft(fft); + + std::complex output_golden[6]; // in ColMajor order + output_golden[0] = std::complex(21, 0); + output_golden[1] = std::complex(-9, 0); + output_golden[2] = std::complex(-3, 1.73205); + output_golden[3] = std::complex( 0, 0); + output_golden[4] = std::complex(-3, -1.73205); + output_golden[5] = std::complex(0 ,0); + + std::complex c_offset = std::complex(1.0, 1.0); + + if (DataLayout == ColMajor) { + VERIFY_IS_APPROX(output(0) + c_offset, output_golden[0] + c_offset); + VERIFY_IS_APPROX(output(1) + c_offset, output_golden[1] + c_offset); + VERIFY_IS_APPROX(output(2) + c_offset, output_golden[2] + c_offset); + VERIFY_IS_APPROX(output(3) + c_offset, output_golden[3] + c_offset); + VERIFY_IS_APPROX(output(4) + c_offset, output_golden[4] + c_offset); + VERIFY_IS_APPROX(output(5) + c_offset, output_golden[5] + c_offset); + } + else { + VERIFY_IS_APPROX(output(0)+ c_offset, output_golden[0]+ c_offset); + VERIFY_IS_APPROX(output(1)+ c_offset, output_golden[2]+ c_offset); + VERIFY_IS_APPROX(output(2)+ c_offset, output_golden[4]+ c_offset); + VERIFY_IS_APPROX(output(3)+ c_offset, output_golden[1]+ c_offset); + VERIFY_IS_APPROX(output(4)+ c_offset, output_golden[3]+ c_offset); + VERIFY_IS_APPROX(output(5)+ c_offset, output_golden[5]+ c_offset); + } +} + +static void test_fft_complex_input_golden() { + Tensor, 1, ColMajor, long> input(5); + input(0) = std::complex(1, 1); + input(1) = std::complex(2, 2); + input(2) = std::complex(3, 3); + input(3) = std::complex(4, 4); + input(4) = std::complex(5, 5); + + array fft; + fft[0] = 0; + + Tensor, 1, ColMajor, long> forward_output_both_parts = input.template fft(fft); + Tensor, 1, ColMajor, long> reverse_output_both_parts = input.template fft(fft); + + Tensor forward_output_real_part = input.template fft(fft); + Tensor reverse_output_real_part = input.template fft(fft); + + Tensor forward_output_imag_part = input.template fft(fft); + Tensor reverse_output_imag_part = input.template fft(fft); + + VERIFY_IS_EQUAL(forward_output_both_parts.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_both_parts.dimension(0), input.dimension(0)); + + VERIFY_IS_EQUAL(forward_output_real_part.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_real_part.dimension(0), input.dimension(0)); + + VERIFY_IS_EQUAL(forward_output_imag_part.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_imag_part.dimension(0), input.dimension(0)); + + std::complex forward_golden_result[5]; + std::complex reverse_golden_result[5]; + + forward_golden_result[0] = std::complex(15.000000000000000,+15.000000000000000); + forward_golden_result[1] = std::complex(-5.940954801177935, +0.940954801177934); + forward_golden_result[2] = std::complex(-3.312299240582266, -1.687700759417735); + forward_golden_result[3] = std::complex(-1.687700759417735, -3.312299240582266); + forward_golden_result[4] = std::complex( 0.940954801177934, -5.940954801177935); + + reverse_golden_result[0] = std::complex( 3.000000000000000, + 3.000000000000000); + reverse_golden_result[1] = std::complex( 0.188190960235587, - 1.188190960235587); + reverse_golden_result[2] = std::complex(-0.337540151883547, - 0.662459848116453); + reverse_golden_result[3] = std::complex(-0.662459848116453, - 0.337540151883547); + reverse_golden_result[4] = std::complex(-1.188190960235587, + 0.188190960235587); + + for(int i = 0; i < 5; ++i) { + VERIFY_IS_APPROX(forward_output_both_parts(i), forward_golden_result[i]); + VERIFY_IS_APPROX(forward_output_real_part(i), forward_golden_result[i].real()); + VERIFY_IS_APPROX(forward_output_imag_part(i), forward_golden_result[i].imag()); + } + + for(int i = 0; i < 5; ++i) { + VERIFY_IS_APPROX(reverse_output_both_parts(i), reverse_golden_result[i]); + VERIFY_IS_APPROX(reverse_output_real_part(i), reverse_golden_result[i].real()); + VERIFY_IS_APPROX(reverse_output_imag_part(i), reverse_golden_result[i].imag()); + } +} + +static void test_fft_real_input_golden() { + Tensor input(5); + input(0) = 1.0; + input(1) = 2.0; + input(2) = 3.0; + input(3) = 4.0; + input(4) = 5.0; + + array fft; + fft[0] = 0; + + Tensor, 1, ColMajor, long> forward_output_both_parts = input.template fft(fft); + Tensor, 1, ColMajor, long> reverse_output_both_parts = input.template fft(fft); + + Tensor forward_output_real_part = input.template fft(fft); + Tensor reverse_output_real_part = input.template fft(fft); + + Tensor forward_output_imag_part = input.template fft(fft); + Tensor reverse_output_imag_part = input.template fft(fft); + + VERIFY_IS_EQUAL(forward_output_both_parts.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_both_parts.dimension(0), input.dimension(0)); + + VERIFY_IS_EQUAL(forward_output_real_part.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_real_part.dimension(0), input.dimension(0)); + + VERIFY_IS_EQUAL(forward_output_imag_part.dimension(0), input.dimension(0)); + VERIFY_IS_EQUAL(reverse_output_imag_part.dimension(0), input.dimension(0)); + + std::complex forward_golden_result[5]; + std::complex reverse_golden_result[5]; + + + forward_golden_result[0] = std::complex( 15, 0); + forward_golden_result[1] = std::complex(-2.5, +3.44095480117793); + forward_golden_result[2] = std::complex(-2.5, +0.81229924058227); + forward_golden_result[3] = std::complex(-2.5, -0.81229924058227); + forward_golden_result[4] = std::complex(-2.5, -3.44095480117793); + + reverse_golden_result[0] = std::complex( 3.0, 0); + reverse_golden_result[1] = std::complex(-0.5, -0.688190960235587); + reverse_golden_result[2] = std::complex(-0.5, -0.162459848116453); + reverse_golden_result[3] = std::complex(-0.5, +0.162459848116453); + reverse_golden_result[4] = std::complex(-0.5, +0.688190960235587); + + std::complex c_offset(1.0, 1.0); + float r_offset = 1.0; + + for(int i = 0; i < 5; ++i) { + VERIFY_IS_APPROX(forward_output_both_parts(i) + c_offset, forward_golden_result[i] + c_offset); + VERIFY_IS_APPROX(forward_output_real_part(i) + r_offset, forward_golden_result[i].real() + r_offset); + VERIFY_IS_APPROX(forward_output_imag_part(i) + r_offset, forward_golden_result[i].imag() + r_offset); + } + + for(int i = 0; i < 5; ++i) { + VERIFY_IS_APPROX(reverse_output_both_parts(i) + c_offset, reverse_golden_result[i] + c_offset); + VERIFY_IS_APPROX(reverse_output_real_part(i) + r_offset, reverse_golden_result[i].real() + r_offset); + VERIFY_IS_APPROX(reverse_output_imag_part(i) + r_offset, reverse_golden_result[i].imag() + r_offset); + } +} + + +template +static void test_fft_real_input_energy() { + + Eigen::DSizes dimensions; + int total_size = 1; + for (int i = 0; i < TensorRank; ++i) { + dimensions[i] = rand() % 20 + 1; + total_size *= dimensions[i]; + } + const DSizes arr = dimensions; + + typedef typename internal::conditional, RealScalar>::type InputScalar; + + Tensor input; + input.resize(arr); + input.setRandom(); + + array fft; + for (int i = 0; i < TensorRank; ++i) { + fft[i] = i; + } + + typedef typename internal::conditional, RealScalar>::type OutputScalar; + Tensor output; + output = input.template fft(fft); + + for (int i = 0; i < TensorRank; ++i) { + VERIFY_IS_EQUAL(output.dimension(i), input.dimension(i)); + } + + float energy_original = 0.0; + float energy_after_fft = 0.0; + + for (int i = 0; i < total_size; ++i) { + energy_original += pow(std::abs(input(i)), 2); + } + + for (int i = 0; i < total_size; ++i) { + energy_after_fft += pow(std::abs(output(i)), 2); + } + + if(FFTDirection == FFT_FORWARD) { + VERIFY_IS_APPROX(energy_original, energy_after_fft / total_size); + } + else { + VERIFY_IS_APPROX(energy_original, energy_after_fft * total_size); + } +} + +void test_cxx11_tensor_fft() { + test_fft_complex_input_golden(); + test_fft_real_input_golden(); + + test_fft_2D_golden(); + test_fft_2D_golden(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); + test_fft_real_input_energy(); +} diff --git a/unsupported/test/cxx11_tensor_ifft.cpp b/unsupported/test/cxx11_tensor_ifft.cpp new file mode 100644 index 000000000..5fd88fa6c --- /dev/null +++ b/unsupported/test/cxx11_tensor_ifft.cpp @@ -0,0 +1,154 @@ +// This file is part of Eigen, a lightweight C++ template library +// for linear algebra. +// +// Copyright (C) 2014 Jianwei Cui +// +// This Source Code Form is subject to the terms of the Mozilla +// Public License v. 2.0. If a copy of the MPL was not distributed +// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. + +#include "main.h" +#include +#include +#include + +using Eigen::Tensor; + +template +static void test_1D_fft_ifft_invariant(int sequence_length) { + Tensor tensor(sequence_length); + tensor.setRandom(); + + array fft; + fft[0] = 0; + + Tensor, 1, DataLayout> tensor_after_fft; + Tensor, 1, DataLayout> tensor_after_fft_ifft; + + tensor_after_fft = tensor.template fft(fft); + tensor_after_fft_ifft = tensor_after_fft.template fft(fft); + + VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), sequence_length); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), sequence_length); + + for (int i = 0; i < sequence_length; ++i) { + VERIFY_IS_APPROX(static_cast(tensor(i)), static_cast(std::real(tensor_after_fft_ifft(i)))); + } +} + +template +static void test_2D_fft_ifft_invariant(int dim0, int dim1) { + Tensor tensor(dim0, dim1); + tensor.setRandom(); + + array fft; + fft[0] = 0; + fft[1] = 1; + + Tensor, 2, DataLayout> tensor_after_fft; + Tensor, 2, DataLayout> tensor_after_fft_ifft; + + tensor_after_fft = tensor.template fft(fft); + tensor_after_fft_ifft = tensor_after_fft.template fft(fft); + + VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(1), dim1); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(1), dim1); + + for (int i = 0; i < dim0; ++i) { + for (int j = 0; j < dim1; ++j) { + //std::cout << "[" << i << "][" << j << "]" << " Original data: " << tensor(i,j) << " Transformed data:" << tensor_after_fft_ifft(i,j) << std::endl; + VERIFY_IS_APPROX(static_cast(tensor(i,j)), static_cast(std::real(tensor_after_fft_ifft(i,j)))); + } + } +} + +template +static void test_3D_fft_ifft_invariant(int dim0, int dim1, int dim2) { + Tensor tensor(dim0, dim1, dim2); + tensor.setRandom(); + + array fft; + fft[0] = 0; + fft[1] = 1; + fft[2] = 2; + + Tensor, 3, DataLayout> tensor_after_fft; + Tensor, 3, DataLayout> tensor_after_fft_ifft; + + tensor_after_fft = tensor.template fft(fft); + tensor_after_fft_ifft = tensor_after_fft.template fft(fft); + + VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(1), dim1); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(2), dim2); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(1), dim1); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(2), dim2); + + for (int i = 0; i < dim0; ++i) { + for (int j = 0; j < dim1; ++j) { + for (int k = 0; k < dim2; ++k) { + VERIFY_IS_APPROX(static_cast(tensor(i,j,k)), static_cast(std::real(tensor_after_fft_ifft(i,j,k)))); + } + } + } +} + +template +static void test_sub_fft_ifft_invariant(int dim0, int dim1, int dim2, int dim3) { + Tensor tensor(dim0, dim1, dim2, dim3); + tensor.setRandom(); + + array fft; + fft[0] = 2; + fft[1] = 0; + + Tensor, 4, DataLayout> tensor_after_fft; + Tensor tensor_after_fft_ifft; + + tensor_after_fft = tensor.template fft(fft); + tensor_after_fft_ifft = tensor_after_fft.template fft(fft); + + VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(1), dim1); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(2), dim2); + VERIFY_IS_EQUAL(tensor_after_fft.dimension(3), dim3); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), dim0); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(1), dim1); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(2), dim2); + VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(3), dim3); + + for (int i = 0; i < dim0; ++i) { + for (int j = 0; j < dim1; ++j) { + for (int k = 0; k < dim2; ++k) { + for (int l = 0; l < dim3; ++l) { + VERIFY_IS_APPROX(static_cast(tensor(i,j,k,l)), static_cast(tensor_after_fft_ifft(i,j,k,l))); + } + } + } + } +} + +void test_cxx11_tensor_ifft() { + CALL_SUBTEST(test_1D_fft_ifft_invariant(4)); + CALL_SUBTEST(test_1D_fft_ifft_invariant(16)); + CALL_SUBTEST(test_1D_fft_ifft_invariant(32)); + CALL_SUBTEST(test_1D_fft_ifft_invariant(1024*1024)); + + CALL_SUBTEST(test_2D_fft_ifft_invariant(4,4)); + CALL_SUBTEST(test_2D_fft_ifft_invariant(8,16)); + CALL_SUBTEST(test_2D_fft_ifft_invariant(16,32)); + CALL_SUBTEST(test_2D_fft_ifft_invariant(1024,1024)); + + CALL_SUBTEST(test_3D_fft_ifft_invariant(4,4,4)); + CALL_SUBTEST(test_3D_fft_ifft_invariant(8,16,32)); + CALL_SUBTEST(test_3D_fft_ifft_invariant(16,4,8)); + CALL_SUBTEST(test_3D_fft_ifft_invariant(256,256,256)); + + CALL_SUBTEST(test_sub_fft_ifft_invariant(4,4,4,4)); + CALL_SUBTEST(test_sub_fft_ifft_invariant(8,16,32,64)); + CALL_SUBTEST(test_sub_fft_ifft_invariant(16,4,8,12)); + CALL_SUBTEST(test_sub_fft_ifft_invariant(64,64,64,64)); +}