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

Finishing touches on igamma/igammac for GPU. Tests now pass.

Browse Source
This commit is contained in:
Eugene Brevdo
2016-03-07 15:35:09 -08:00
parent 5707004d6b
commit 0bb5de05a1
2 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
unsupported/test/cxx11_tensor_cuda.cu
View File

@@ -632,7 +632,6 @@ void test_cuda_igamma()
Tensor<Scalar, 2> a(6, 6);
Tensor<Scalar, 2> x(6, 6);
Tensor<Scalar, 2> out(6, 6);
Tensor<Scalar, 2> expected_out(6, 6);
out.setZero();
Scalar a_s[] = {Scalar(0), Scalar(1), Scalar(1.5), Scalar(4), Scalar(0.0001), Scalar(1000.5)};
@@ -641,7 +640,7 @@ void test_cuda_igamma()
for (int i = 0; i < 6; ++i) {
for (int j = 0; j < 6; ++j) {
a(i, j) = a_s[i];
x(i, j) = x_s[i];
x(i, j) = x_s[j];
}
}
@@ -686,8 +685,7 @@ void test_cuda_igamma()
for (int i = 0; i < 6; ++i) {
for (int j = 0; j < 6; ++j) {
if ((std::isnan)(igamma_s[i][j])) {
printf("got: %g\n", out(i, j));
//VERIFY((std::isnan)(out(i, j)));
VERIFY((std::isnan)(out(i, j)));
} else {
VERIFY_IS_APPROX(out(i, j), igamma_s[i][j]);
}
@@ -701,7 +699,6 @@ void test_cuda_igammac()
Tensor<Scalar, 2> a(6, 6);
Tensor<Scalar, 2> x(6, 6);
Tensor<Scalar, 2> out(6, 6);
Tensor<Scalar, 2> expected_out(6, 6);
out.setZero();
Scalar a_s[] = {Scalar(0), Scalar(1), Scalar(1.5), Scalar(4), Scalar(0.0001), Scalar(1000.5)};
@@ -710,7 +707,7 @@ void test_cuda_igammac()
for (int i = 0; i < 6; ++i) {
for (int j = 0; j < 6; ++j) {
a(i, j) = a_s[i];
x(i, j) = x_s[i];
x(i, j) = x_s[j];
}
}
@@ -754,8 +751,7 @@ void test_cuda_igammac()
for (int i = 0; i < 6; ++i) {
for (int j = 0; j < 6; ++j) {
if ((std::isnan)(igammac_s[i][j])) {
printf("got: %g\n", out(i, j));
//VERIFY((std::isnan)(out(i, j)));
VERIFY((std::isnan)(out(i, j)));
} else {
VERIFY_IS_APPROX(out(i, j), igammac_s[i][j]);
}