devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Created additional unit tests for the tensor code and improved existing ones.

Browse Source
This commit is contained in:
Benoit Steiner
2014-06-13 10:20:28 -07:00
parent f80c8e17eb
commit 774c3c1e0a
4 changed files with 143 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
unsupported/test/cxx11_tensor_device.cpp
View File

@@ -65,6 +65,12 @@ static void test_contextual_eval(Context* context)
context->out() = context->in1() + context->in2() * 3.14f + context->in1().constant(2.718f);
}
template <typename Context>
static void test_forced_contextual_eval(Context* context)
{
context->out() = (context->in1() + context->in2()).eval() * 3.14f + context->in1().constant(2.718f);
}
static void test_cpu() {
Eigen::Tensor<float, 3> in1(Eigen::array<int, 3>(2,3,7));
Eigen::Tensor<float, 3> in2(Eigen::array<int, 3>(2,3,7));
@@ -72,9 +78,9 @@ static void test_cpu() {
in1.setRandom();
in2.setRandom();
CPUContext context(in1, in2, out);
test_contextual_eval(&context);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
@@ -82,6 +88,15 @@ static void test_cpu() {
}
}
}
test_forced_contextual_eval(&context);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_APPROX(out(Eigen::array<int, 3>(i,j,k)), (in1(Eigen::array<int, 3>(i,j,k)) + in2(Eigen::array<int, 3>(i,j,k))) * 3.14f + 2.718f);
}
}
}
}
static void test_gpu() {
@@ -111,7 +126,6 @@ static void test_gpu() {
GPUContext context(gpu_in1, gpu_in2, gpu_out);
test_contextual_eval(&context);
cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
@@ -120,6 +134,16 @@ static void test_gpu() {
}
}
}
test_forced_contextual_eval(&context);
cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost);
for (int i = 0; i < 2; ++i) {
for (int j = 0; j < 3; ++j) {
for (int k = 0; k < 7; ++k) {
VERIFY_IS_APPROX(out(Eigen::array<int, 3>(i,j,k)), (in1(Eigen::array<int, 3>(i,j,k)) + in2(Eigen::array<int, 3>(i,j,k))) * 3.14f + 2.718f);
}
}
}
}