diff options
Diffstat (limited to 'unsupported/test/cxx11_tensor_of_float16_gpu.cu')
-rw-r--r-- | unsupported/test/cxx11_tensor_of_float16_gpu.cu | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/unsupported/test/cxx11_tensor_of_float16_gpu.cu b/unsupported/test/cxx11_tensor_of_float16_gpu.cu index 4d74e6138..c55676c76 100644 --- a/unsupported/test/cxx11_tensor_of_float16_gpu.cu +++ b/unsupported/test/cxx11_tensor_of_float16_gpu.cu @@ -64,7 +64,7 @@ void test_gpu_conversion() { Eigen::GpuStreamDevice stream; Eigen::GpuDevice gpu_device(&stream); int num_elem = 101; - + float* d_float = (float*)gpu_device.allocate(num_elem * sizeof(float)); 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)); @@ -322,7 +322,7 @@ template<typename> void test_gpu_reductions(int size1, int size2, int redux) { std::cout << "Reducing " << size1 << " by " << size2 - << " tensor along dim " << redux << std::endl; + << " tensor along dim " << redux << std::endl; Eigen::GpuStreamDevice stream; Eigen::GpuDevice gpu_device(&stream); @@ -346,7 +346,7 @@ void test_gpu_reductions(int size1, int size2, int redux) { gpu_float1.device(gpu_device) = gpu_float1.random() * 2.0f; gpu_float2.device(gpu_device) = gpu_float2.random() * 2.0f; - Eigen::array<int, 1> redux_dim = {{redux}}; + Eigen::array<int, 1> redux_dim = {redux}; gpu_res_float.device(gpu_device) = gpu_float1.sum(redux_dim).cast<Eigen::half>(); gpu_res_half.device(gpu_device) = gpu_float1.cast<Eigen::half>().sum(redux_dim); |