From 0606a0a39bcf01b0a03f0dcd17f7075fce8c402c Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Thu, 18 Feb 2016 23:15:23 -0800 Subject: FP16 on CUDA are only available starting with cuda 7.5. Disable them when using an older version of CUDA --- unsupported/test/cxx11_tensor_of_float16_cuda.cu | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'unsupported') diff --git a/unsupported/test/cxx11_tensor_of_float16_cuda.cu b/unsupported/test/cxx11_tensor_of_float16_cuda.cu index e9f5dd968..aee222a14 100644 --- a/unsupported/test/cxx11_tensor_of_float16_cuda.cu +++ b/unsupported/test/cxx11_tensor_of_float16_cuda.cu @@ -19,6 +19,7 @@ using Eigen::Tensor; +#ifdef EIGEN_HAS_CUDA_FP16 void test_cuda_conversion() { Eigen::CudaStreamDevice stream; Eigen::GpuDevice gpu_device(&stream); @@ -52,9 +53,11 @@ void test_cuda_conversion() { gpu_device.deallocate(d_half); gpu_device.deallocate(d_conv); } - +#endif void test_cxx11_tensor_of_float16_cuda() { +#ifdef EIGEN_HAS_CUDA_FP16 CALL_SUBTEST_1(test_cuda_conversion()); +#endif } -- cgit v1.2.3