diff --git a/unsupported/test/cxx11_tensor_of_float16_cuda.cu b/unsupported/test/cxx11_tensor_of_float16_cuda.cu index 9b9fd843c..29b5637e7 100644 --- a/unsupported/test/cxx11_tensor_of_float16_cuda.cu +++ b/unsupported/test/cxx11_tensor_of_float16_cuda.cu @@ -27,18 +27,18 @@ void test_cuda_conversion() { int num_elem = 101; float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); - half* d_half = (half*)gpu_device.allocate(num_elem * sizeof(half)); + Eigen::half* d_half = (Eigen::half*)gpu_device.allocate(num_elem * sizeof(Eigen::half)); float* d_conv = (float*)gpu_device.allocate(num_elem * sizeof(float)); Eigen::TensorMap, Eigen::Aligned> gpu_float( d_float, num_elem); - Eigen::TensorMap, Eigen::Aligned> gpu_half( + Eigen::TensorMap, Eigen::Aligned> gpu_half( d_half, num_elem); Eigen::TensorMap, Eigen::Aligned> gpu_conv( d_conv, num_elem); gpu_float.device(gpu_device) = gpu_float.random(); - gpu_half.device(gpu_device) = gpu_float.cast(); + gpu_half.device(gpu_device) = gpu_float.cast(); gpu_conv.device(gpu_device) = gpu_half.cast(); Tensor initial(num_elem); @@ -72,9 +72,9 @@ void test_cuda_unary() { Eigen::TensorMap, Eigen::Aligned> gpu_res_float( d_res_float, num_elem); - gpu_float.device(gpu_device) = gpu_float.random(); + gpu_float.device(gpu_device) = gpu_float.random() - gpu_float.constant(0.5f); gpu_res_float.device(gpu_device) = gpu_float.abs(); - gpu_res_half.device(gpu_device) = gpu_float.cast().abs().cast(); + gpu_res_half.device(gpu_device) = gpu_float.cast().abs().cast(); Tensor half_prec(num_elem); Tensor full_prec(num_elem); @@ -115,7 +115,7 @@ void test_cuda_elementwise() { gpu_float1.device(gpu_device) = gpu_float1.random(); gpu_float2.device(gpu_device) = gpu_float2.random(); gpu_res_float.device(gpu_device) = (gpu_float1 + gpu_float2) * gpu_float1; - gpu_res_half.device(gpu_device) = ((gpu_float1.cast() + gpu_float2.cast()) * gpu_float1.cast()).cast(); + gpu_res_half.device(gpu_device) = ((gpu_float1.cast() + gpu_float2.cast()) * gpu_float1.cast()).cast(); Tensor half_prec(num_elem); Tensor full_prec(num_elem); @@ -162,7 +162,7 @@ void test_cuda_contractions() { typedef Tensor::DimensionPair DimPair; Eigen::array dims(DimPair(1, 0)); gpu_res_float.device(gpu_device) = gpu_float1.contract(gpu_float2, dims); - gpu_res_half.device(gpu_device) = gpu_float1.cast().contract(gpu_float2.cast(), dims).cast(); + gpu_res_half.device(gpu_device) = gpu_float1.cast().contract(gpu_float2.cast(), dims).cast(); Tensor half_prec(rows, cols); Tensor full_prec(rows, cols); @@ -209,7 +209,7 @@ void test_cuda_reductions() { Eigen::array redux_dim = {{0}}; gpu_res_float.device(gpu_device) = gpu_float1.sum(redux_dim); - gpu_res_half.device(gpu_device) = gpu_float1.cast().sum(redux_dim).cast(); + gpu_res_half.device(gpu_device) = gpu_float1.cast().sum(redux_dim).cast(); Tensor half_prec(size); Tensor full_prec(size);