diff options
author | Mehdi Goli <mehdi.goli@codeplay.com> | 2016-11-14 17:51:57 +0000 |
---|---|---|
committer | Mehdi Goli <mehdi.goli@codeplay.com> | 2016-11-14 17:51:57 +0000 |
commit | f8ca893976316df46791d2f088552fb2aea419bb (patch) | |
tree | 85a25dfa9e9e669334f5120e8085e70f1b2e3a3e /unsupported/test/cxx11_tensor_broadcast_sycl.cpp | |
parent | a5c3f15682299495f98b6f5480c798fd3211f590 (diff) |
Adding TensorFixsize; adding sycl device memcpy; adding insial stage of slicing.
Diffstat (limited to 'unsupported/test/cxx11_tensor_broadcast_sycl.cpp')
-rw-r--r-- | unsupported/test/cxx11_tensor_broadcast_sycl.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/unsupported/test/cxx11_tensor_broadcast_sycl.cpp b/unsupported/test/cxx11_tensor_broadcast_sycl.cpp index 7201bfe37..11b608ea4 100644 --- a/unsupported/test/cxx11_tensor_broadcast_sycl.cpp +++ b/unsupported/test/cxx11_tensor_broadcast_sycl.cpp @@ -47,7 +47,8 @@ static void test_broadcast_sycl(const Eigen::SyclDevice &sycl_device){ float * gpu_in_data = static_cast<float*>(sycl_device.allocate(input.dimensions().TotalSize()*sizeof(float))); float * gpu_out_data = static_cast<float*>(sycl_device.allocate(out.dimensions().TotalSize()*sizeof(float))); - TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range); + TensorMap<TensorFixedSize<float, Sizes<2, 3, 5, 7>>> gpu_in(gpu_in_data, in_range); + //TensorMap<Tensor<float, 4>> gpu_in(gpu_in_data, in_range); TensorMap<Tensor<float, 4>> gpu_out(gpu_out_data, out_range); sycl_device.memcpyHostToDevice(gpu_in_data, input.data(),(input.dimensions().TotalSize())*sizeof(float)); gpu_out.device(sycl_device) = gpu_in.broadcast(broadcasts); |