diff --git a/unsupported/test/cxx11_tensor_cast_float16_cuda.cu b/unsupported/test/cxx11_tensor_cast_float16_cuda.cu index d9ed4c855..fece57482 100644 --- a/unsupported/test/cxx11_tensor_cast_float16_cuda.cu +++ b/unsupported/test/cxx11_tensor_cast_float16_cuda.cu @@ -59,7 +59,22 @@ void test_cuda_conversion() { } +void test_fallback_conversion() { + int num_elem = 101; + Tensor floats(num_elem); + floats.setRandom(); + + Eigen::Tensor halfs = floats.cast(); + Eigen::Tensor conv = half.cast(); + + for (int i = 0; i < num_elem; ++i) { + VERIFY_IS_APPROX(floats(i), conv(i)); + } +} + + void test_cxx11_tensor_cast_float16_cuda() { CALL_SUBTEST(test_cuda_conversion()); + CALL_SUBTEST(test_fallback_conversion()); }