Home
last modified time | relevance | path

Searched refs:redux_gpu (Results 1 – 2 of 2) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction_sycl.cpp68 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_first_dim_reductions_sycl() local
75 …float* gpu_out_data = static_cast<float*>(sycl_device.allocate(redux_gpu.dimensions().TotalSize()*… in test_first_dim_reductions_sycl()
82 …sycl_device.memcpyDeviceToHost(redux_gpu.data(), gpu_out_data, redux_gpu.dimensions().TotalSize()*… in test_first_dim_reductions_sycl()
87 VERIFY_IS_APPROX(redux_gpu(j,k), redux(j,k)); in test_first_dim_reductions_sycl()
106 Tensor<float, 2> redux_gpu(reduced_tensorRange); in test_last_dim_reductions_sycl() local
113 …float* gpu_out_data = static_cast<float*>(sycl_device.allocate(redux_gpu.dimensions().TotalSize()*… in test_last_dim_reductions_sycl()
120 …sycl_device.memcpyDeviceToHost(redux_gpu.data(), gpu_out_data, redux_gpu.dimensions().TotalSize()*… in test_last_dim_reductions_sycl()
124 VERIFY_IS_APPROX(redux_gpu(j,k), redux(j,k)); in test_last_dim_reductions_sycl()
Dcxx11_tensor_reduction_cuda.cu86 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions() local
87 dev.memcpyDeviceToHost(redux_gpu.data(), out_data, gpu_out.size()*sizeof(Type)); in test_first_dim_reductions()
92 VERIFY_IS_APPROX(2*redux(i), redux_gpu(i)); in test_first_dim_reductions()
126 Tensor<Type, 2, DataLayout> redux_gpu(dim_x, dim_y); in test_last_dim_reductions() local
127 dev.memcpyDeviceToHost(redux_gpu.data(), out_data, gpu_out.size()*sizeof(Type)); in test_last_dim_reductions()
132 VERIFY_IS_APPROX(2*redux(i), redux_gpu(i)); in test_last_dim_reductions()