GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-24 02:29:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleaned up the new float16 test a bit

Browse Source
This commit is contained in:
Benoit Steiner 2016-08-03 11:50:07 -07:00
parent 81099ef482
commit d92df04ce8
1 changed files with 24 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
unsupported/test/cxx11_float16.cpp
View File

@ -11,12 +11,14 @@
#define EIGEN_DEFAULT_DENSE_INDEX_TYPE int #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int
#define EIGEN_USE_GPU #define EIGEN_USE_GPU
// Make sure it's possible to forward declare Eigen::half
//namespace Eigen {
//struct half;
//}
#include "main.h" #include "main.h"
#include <Eigen/src/Core/arch/CUDA/Half.h> #include <Eigen/src/Core/arch/CUDA/Half.h>
using Eigen::half;
void test_conversion() void test_conversion()
{ {
// Conversion from float. // Conversion from float.
@ -37,8 +39,8 @@ void test_conversion()
float val1 = float(Eigen::half(half_impl::__half{0x3c00})); float val1 = float(Eigen::half(half_impl::__half{0x3c00}));
float val2 = float(Eigen::half(half_impl::__half{0x3c01})); float val2 = float(Eigen::half(half_impl::__half{0x3c01}));
float val3 = float(Eigen::half(half_impl::__half{0x3c02})); float val3 = float(Eigen::half(half_impl::__half{0x3c02}));
VERIFY_IS_EQUAL(Eigen::half(0.5 * (val1 + val2)).x, 0x3c00); VERIFY_IS_EQUAL(Eigen::half(0.5f * (val1 + val2)).x, 0x3c00);
VERIFY_IS_EQUAL(Eigen::half(0.5 * (val2 + val3)).x, 0x3c02); VERIFY_IS_EQUAL(Eigen::half(0.5f * (val2 + val3)).x, 0x3c02);
// Conversion from int. // Conversion from int.
VERIFY_IS_EQUAL(Eigen::half(-1).x, 0xbc00); VERIFY_IS_EQUAL(Eigen::half(-1).x, 0xbc00);
@ -61,26 +63,26 @@ void test_conversion()
VERIFY_IS_APPROX(float(Eigen::half(half_impl::__half{0x0002})), 1.19209e-07f); VERIFY_IS_APPROX(float(Eigen::half(half_impl::__half{0x0002})), 1.19209e-07f);
// NaNs and infinities. // NaNs and infinities.
VERIFY(!isinf(float(Eigen::half(65504.0f)))); // Largest finite number. VERIFY(!(isinf)(float(Eigen::half(65504.0f)))); // Largest finite number.
VERIFY(!isnan(float(Eigen::half(0.0f)))); VERIFY(!(isnan)(float(Eigen::half(0.0f))));
VERIFY(isinf(float(Eigen::half(half_impl::__half{0xfc00})))); VERIFY((isinf)(float(Eigen::half(half_impl::__half{0xfc00}))));
VERIFY(isnan(float(Eigen::half(half_impl::__half{0xfc01})))); VERIFY((isnan)(float(Eigen::half(half_impl::__half{0xfc01}))));
VERIFY(isinf(float(Eigen::half(half_impl::__half{0x7c00})))); VERIFY((isinf)(float(Eigen::half(half_impl::__half{0x7c00}))));
VERIFY(isnan(float(Eigen::half(half_impl::__half{0x7c01})))); VERIFY((isnan)(float(Eigen::half(half_impl::__half{0x7c01}))));
VERIFY(isnan(float(Eigen::half(0.0 / 0.0)))); VERIFY((isnan)(float(Eigen::half(0.0 / 0.0))));
VERIFY(isinf(float(Eigen::half(1.0 / 0.0)))); VERIFY((isinf)(float(Eigen::half(1.0 / 0.0))));
VERIFY(isinf(float(Eigen::half(-1.0 / 0.0)))); VERIFY((isinf)(float(Eigen::half(-1.0 / 0.0))));
// Exactly same checks as above, just directly on the half representation. // Exactly same checks as above, just directly on the half representation.
VERIFY(!numext::isinf(Eigen::half(half_impl::__half{0x7bff}))); VERIFY(!(numext::isinf)(Eigen::half(half_impl::__half{0x7bff})));
VERIFY(!numext::isnan(Eigen::half(half_impl::__half{0x0000}))); VERIFY(!(numext::isnan)(Eigen::half(half_impl::__half{0x0000})));
VERIFY(numext::isinf(Eigen::half(half_impl::__half{0xfc00}))); VERIFY((numext::isinf)(Eigen::half(half_impl::__half{0xfc00})));
VERIFY(numext::isnan(Eigen::half(half_impl::__half{0xfc01}))); VERIFY((numext::isnan)(Eigen::half(half_impl::__half{0xfc01})));
VERIFY(numext::isinf(Eigen::half(half_impl::__half{0x7c00}))); VERIFY((numext::isinf)(Eigen::half(half_impl::__half{0x7c00})));
VERIFY(numext::isnan(Eigen::half(half_impl::__half{0x7c01}))); VERIFY((numext::isnan)(Eigen::half(half_impl::__half{0x7c01})));
VERIFY(numext::isnan(Eigen::half(0.0 / 0.0))); VERIFY((numext::isnan)(Eigen::half(0.0 / 0.0)));
VERIFY(numext::isinf(Eigen::half(1.0 / 0.0))); VERIFY((numext::isinf)(Eigen::half(1.0 / 0.0)));
VERIFY(numext::isinf(Eigen::half(-1.0 / 0.0))); VERIFY((numext::isinf)(Eigen::half(-1.0 / 0.0)));
} }
void test_arithmetic() void test_arithmetic()