devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merging eigen/eigen.

Browse Source
This commit is contained in:
Srinivas Vasudevan
2019-09-16 19:33:29 -04:00
parent 6e215cf109 7c73296849
commit df0816b71f
12 changed files with 507 additions and 321 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/test/cxx11_tensor_gpu.cu
View File

@@ -1317,7 +1317,7 @@ void test_gpu_i0e()
Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 21);
Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 21);
gpu_out.device(gpu_device) = gpu_in.i0e();
gpu_out.device(gpu_device) = gpu_in.bessel_i0e();
assert(gpuMemcpyAsync(out.data(), d_out, bytes, gpuMemcpyDeviceToHost,
gpu_device.stream()) == gpuSuccess);
@@ -1372,7 +1372,7 @@ void test_gpu_i1e()
Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_in(d_in, 21);
Eigen::TensorMap<Eigen::Tensor<Scalar, 1> > gpu_out(d_out, 21);
gpu_out.device(gpu_device) = gpu_in.i1e();
gpu_out.device(gpu_device) = gpu_in.bessel_i1e();
assert(gpuMemcpyAsync(out.data(), d_out, bytes, gpuMemcpyDeviceToHost,
gpu_device.stream()) == gpuSuccess);