diff options
author | Teng Lu <teng.lu@intel.com> | 2020-06-20 19:16:24 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-06-20 19:16:24 +0000 |
commit | 386d809bde475c65b7940f290efe80e6a05878c4 (patch) | |
tree | c38e161a53393d15be0ddb02a7a4e22dec738484 /unsupported/test | |
parent | 6b9c92fe7eff0dedb031cec38004c9c3667f3057 (diff) |
Support BFloat16 in Eigen
Diffstat (limited to 'unsupported/test')
-rw-r--r-- | unsupported/test/cxx11_tensor_reduction.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/unsupported/test/cxx11_tensor_reduction.cpp b/unsupported/test/cxx11_tensor_reduction.cpp index 996dba806..f1ac83b1b 100644 --- a/unsupported/test/cxx11_tensor_reduction.cpp +++ b/unsupported/test/cxx11_tensor_reduction.cpp @@ -511,6 +511,7 @@ EIGEN_DECLARE_TEST(cxx11_tensor_reduction) { CALL_SUBTEST(( test_simple_reductions<float,ColMajor>() )); CALL_SUBTEST(( test_simple_reductions<float,RowMajor>() )); CALL_SUBTEST(( test_simple_reductions<Eigen::half,ColMajor>() )); + CALL_SUBTEST(( test_simple_reductions<Eigen::bfloat16,ColMajor>() )); CALL_SUBTEST(test_reductions_in_expr<ColMajor>()); CALL_SUBTEST(test_reductions_in_expr<RowMajor>()); CALL_SUBTEST(test_full_reductions<ColMajor>()); |