Home
last modified time | relevance | path

Searched refs:sycl_device (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction_sycl.cpp25 static void test_full_reductions_sycl(const Eigen::SyclDevice& sycl_device) { in test_full_reductions_sycl() argument
39 …float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(f… in test_full_reductions_sycl()
40 float* gpu_out_data =(float*)sycl_device.allocate(sizeof(float)); in test_full_reductions_sycl()
45 …sycl_device.memcpyHostToDevice(gpu_in_data, in.data(),(in.dimensions().TotalSize())*sizeof(float)); in test_full_reductions_sycl()
46 out_gpu.device(sycl_device) = in_gpu.sum(); in test_full_reductions_sycl()
47 sycl_device.memcpyDeviceToHost(full_redux_gpu.data(), gpu_out_data, sizeof(float)); in test_full_reductions_sycl()
51 sycl_device.deallocate(gpu_in_data); in test_full_reductions_sycl()
52 sycl_device.deallocate(gpu_out_data); in test_full_reductions_sycl()
55 static void test_first_dim_reductions_sycl(const Eigen::SyclDevice& sycl_device) { in test_first_dim_reductions_sycl() argument
74 …float* gpu_in_data = static_cast<float*>(sycl_device.allocate(in.dimensions().TotalSize()*sizeof(f… in test_first_dim_reductions_sycl()
[all …]
Dcxx11_tensor_sycl.cpp30 void test_sycl_cpu(const Eigen::SyclDevice &sycl_device) { in test_sycl_cpu() argument
44 …float * gpu_in1_data = static_cast<float*>(sycl_device.allocate(in1.dimensions().TotalSize()*size… in test_sycl_cpu()
45 …float * gpu_in2_data = static_cast<float*>(sycl_device.allocate(in2.dimensions().TotalSize()*size… in test_sycl_cpu()
46 …float * gpu_in3_data = static_cast<float*>(sycl_device.allocate(in3.dimensions().TotalSize()*size… in test_sycl_cpu()
47 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_sycl_cpu()
55 gpu_in1.device(sycl_device) = gpu_in1.constant(1.2f); in test_sycl_cpu()
56 …sycl_device.memcpyDeviceToHost(in1.data(), gpu_in1_data ,(in1.dimensions().TotalSize())*sizeof(flo… in test_sycl_cpu()
67 gpu_out.device(sycl_device) = gpu_in1 * 1.2f; in test_sycl_cpu()
68 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data ,(out.dimensions().TotalSize())*sizeof(flo… in test_sycl_cpu()
80 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in2.dimensions().TotalSize())*sizeof(floa… in test_sycl_cpu()
[all …]
Dcxx11_tensor_forced_eval_sycl.cpp25 void test_forced_eval_sycl(const Eigen::SyclDevice &sycl_device) { in test_forced_eval_sycl() argument
35 …float * gpu_in1_data = static_cast<float*>(sycl_device.allocate(in1.dimensions().TotalSize()*size… in test_forced_eval_sycl()
36 …float * gpu_in2_data = static_cast<float*>(sycl_device.allocate(in2.dimensions().TotalSize()*size… in test_forced_eval_sycl()
37 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_forced_eval_sycl()
46 …sycl_device.memcpyHostToDevice(gpu_in1_data, in1.data(),(in1.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
47 …sycl_device.memcpyHostToDevice(gpu_in2_data, in2.data(),(in1.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
49 gpu_out.device(sycl_device) =(gpu_in1 + gpu_in2).eval() * gpu_in2; in test_forced_eval_sycl()
50 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_forced_eval_sycl()
60 sycl_device.deallocate(gpu_in1_data); in test_forced_eval_sycl()
61 sycl_device.deallocate(gpu_in2_data); in test_forced_eval_sycl()
[all …]
Dcxx11_tensor_broadcast_sycl.cpp28 static void test_broadcast_sycl(const Eigen::SyclDevice &sycl_device){ in test_broadcast_sycl() argument
47 …float * gpu_in_data = static_cast<float*>(sycl_device.allocate(input.dimensions().TotalSize()*siz… in test_broadcast_sycl()
48 …float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*size… in test_broadcast_sycl()
52 …sycl_device.memcpyHostToDevice(gpu_in_data, input.data(),(input.dimensions().TotalSize())*sizeof(f… in test_broadcast_sycl()
53 gpu_out.device(sycl_device) = gpu_in.broadcast(broadcasts); in test_broadcast_sycl()
54 …sycl_device.memcpyDeviceToHost(out.data(), gpu_out_data,(out.dimensions().TotalSize())*sizeof(floa… in test_broadcast_sycl()
66 sycl_device.deallocate(gpu_in_data); in test_broadcast_sycl()
67 sycl_device.deallocate(gpu_out_data); in test_broadcast_sycl()
72 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_broadcast_sycl() local
73 CALL_SUBTEST(test_broadcast_sycl(sycl_device)); in test_cxx11_tensor_broadcast_sycl()
Dcxx11_tensor_device_sycl.cpp23 void test_device_sycl(const Eigen::SyclDevice &sycl_device) { in test_device_sycl() argument
25 …<< sycl_device.m_queue.get_device(). template get_info<cl::sycl::info::device::name>() <<std::endl… in test_device_sycl()
29 Eigen::SyclDevice sycl_device(s); in test_cxx11_tensor_device_sycl() local
30 CALL_SUBTEST(test_device_sycl(sycl_device)); in test_cxx11_tensor_device_sycl()