GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-09-12 17:33:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding synchronisation to convolution kernel for sycl backend.

Browse Source
This commit is contained in:
Mehdi Goli 2017-03-13 09:18:37 +00:00
parent 1b32a10053
commit f499fe9496
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bench/tensors/tensor_benchmarks.h
View File

@ -539,6 +539,11 @@ for (int iter = 0; iter < 10; ++iter) {
if (Eigen::internal::is_same<Device, Eigen::GpuDevice>::value) { if (Eigen::internal::is_same<Device, Eigen::GpuDevice>::value) {
device_.synchronize(); device_.synchronize();
} }
#elif defined(EIGEN_USE_SYCL)
if (Eigen::internal::is_same<Device, Eigen::SyclDevice>::value) {
device_.synchronize();
}
#endif #endif
StopBenchmarkTiming(); StopBenchmarkTiming();
SetBenchmarkFlopsProcessed(num_items); SetBenchmarkFlopsProcessed(num_items);

1
unsupported/Eigen/CXX11/src/Tensor/TensorConvolutionSycl.h
View File

@ -425,6 +425,7 @@ struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelAr
} }
} }
}); });
m_device.asynchronousExec();
} }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index index) const